mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity
424 Commits 11 Branches 5 Tags
4b3731928c63f802a2804924463c57d33cf244d1
Commit Graph

424 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Micaela Verucchi 4b3731928c Add yolo4_320_coco2 (pedestrian and stop sign) 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2021-02-25 09:57:10 +01:00
Micaela Verucchi a17e7800b9 Add computation of #parameters, #MACC, and max feature map size in the tests 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-11-23 13:07:54 +01:00
Micaela Verucchi 04b4a69107 Merge with master 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-09-14 17:30:02 +02:00
micaela fa2b3d26cb Fix typos (#107) 
Signed-off-by: micaela <micaelaverucchi@gmail.com>
 2020-09-11 09:13:59 +02:00
Micaela Verucchi 38106a9495 Merge branch 'master' of https://github.com/ceccocats/tkDNN 2020-08-06 15:54:09 +02:00
Micaela Verucchi df5443e017 Fix boxes also for Centernet 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-08-06 15:53:57 +02:00
Micaela Verucchi 8bef544bae Update README.md 2020-08-06 11:05:26 +02:00
Micaela Verucchi f778e1aa99 Fixed boxes to float, add conf thresh as param 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-08-05 19:55:10 +02:00
Francesco Gatti a5d2d4792a fix coords convert 2020-07-27 13:52:39 +02:00
Francesco Gatti 3a0802d70c Resolve detection objects pick by prob threshold. 
Before this it will only pick the last object with prob > thresh wich is absolutely wrong
Now it picks all the objects with prob > thesh.

fixes #94
 2020-07-27 13:45:42 +02:00
Micaela Verucchi f4970d1e6f Update README.md 2020-07-17 14:37:10 +02:00
Micaela Verucchi 6a68f19b2c Fix patch from @ahmedius2 , tkDNN now supports CUDNN 8.0.1 (Fix #74) 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
	       Francesco Gatti <gattifrancesco@hotmail.it>
 2020-07-16 19:06:54 +02:00
tk c4aad7fe95 Patch for CUDNN 8.0.1 
Signed-off-by: tk <micaelaverucchi@gmail.com>
 2020-07-16 18:16:09 +02:00
Francesco Gatti b2df9fc110 Update README.md 2020-07-13 19:51:09 +02:00
Francesco Gatti b12cf0d7c2 docker 2020-07-13 19:50:00 +02:00
Micaela Verucchi 594947f301 Update file names map demo 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-07-08 11:27:09 +02:00
Micaela Verucchi e7779ad773 Add conf_thresh parameter to detectors, add batches handling in map demo 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-07-08 09:52:36 +02:00
tk 65e2074dda Merge branch 'eval' of https://github.com/ceccocats/tkDNN into eval 2020-07-06 17:58:08 +02:00
tk a68f45cb4e Merge with master, add yolov4tiny_512 
Signed-off-by: tk <micaelaverucchi@gmail.com>
 2020-07-06 17:57:50 +02:00
Micaela Verucchi 7c2155decf Add weights download link for csresnext50-panet-spp_berkeley ( fix #63 ) 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-07-01 11:56:46 +02:00
Micaela Verucchi a4dca23111 Minor 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-07-01 10:22:31 +02:00
Francesco Gatti 04602f3952 Yolo4-tiny batched 
fix #59
 2020-06-30 19:37:16 +02:00
Micaela Verucchi fe2e4eae92 yolo4tiny works on tensorRT 🐬 🐬 🐬 🐬 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-30 15:52:58 +02:00
Micaela Verucchi 61aa24c6b7 yolov4tiny works on CUDNN 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-30 15:19:03 +02:00
Francesco Gatti 6d9beb1ec5 Merge pull request #53 from omaralvarez/master 
Fix error when parsing label files with unexpected chars
 2020-06-25 10:07:41 +02:00
Omar Alvarez 2817ade782 Fix parsing label files with unexpected chars 2020-06-22 13:40:15 +02:00
Francesco Gatti 1dfc69ba89 viz yolo3 preprocess 2020-06-16 13:19:30 +02:00
Francesco Gatti 07193fc343 dealloc in test.h 
fix #36
 2020-06-16 12:44:12 +02:00
Francesco Gatti 1b8f45703f darknet parser cpp 2020-06-16 12:41:48 +02:00
Francesco Gatti 3d3a2427c9 viz yolo3 2020-06-16 12:36:45 +02:00
Micaela Verucchi 285c77d6dd Imptove relative paths 
Signed-off-by:  Micaela Verucchi <micaelaverucchi@gmail.com>
		Francesco Gatti <gattifrancesco@hotmail.it>
 2020-06-15 17:15:37 +02:00
Francesco Gatti 6dff675db7 fix wrong commit 2020-06-14 13:03:48 +02:00
Francesco Gatti cbfc8ea4f2 serialize fix 2020-06-14 13:01:45 +02:00
Francesco Gatti 567dc0f75d cmake cudnn fix 2020-06-14 12:57:29 +02:00
Micaela Verucchi ab6d2d1766 Update README 
Signed-off-by: Micaela Verucchi <micaela.verucchi@unimore.it>
 2020-06-12 11:39:46 +02:00
Micaela Verucchi 8d08f5aade New yolo4_512 download link 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-11 20:54:17 +02:00
Micaela Verucchi e094a3e0fc Add script for inference FPS 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-11 16:09:01 +02:00
Micaela Verucchi c4e955eab5 Add different yolov4 size tests 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-11 12:43:34 +02:00
Micaela Verucchi be9e327aef Fix minor, update readme 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-09 21:01:19 +02:00
Francesco Gatti 211eff8ad0 Merge branch 'master' of https://github.com/ceccocats/tkDNN 2020-06-09 20:53:17 +02:00
Francesco Gatti 20303ac32e cudnn8 compile 2020-06-09 20:53:05 +02:00
Micaela Verucchi 3ea23815a4 Update README.md 2020-06-03 11:01:17 +02:00
Francesco Gatti 62e4a3f779 memory release v0.5 2020-06-02 12:43:06 +02:00
Francesco Gatti 0458f361b1 release layer wgs and version update 2020-06-01 21:03:37 +02:00
Francesco Gatti 2f243f26e5 readme update 2020-06-01 19:04:39 +02:00
Francesco Gatti c8ed6d782a all test ok 2020-06-01 16:15:29 +02:00
Francesco Gatti d4e0d07e09 merge 2020-06-01 15:34:59 +02:00
Micaela Verucchi 0548a662f5 Merge branch 'master' of https://github.com/ceccocats/tkDNN into eval 2020-06-01 15:18:29 +02:00
Micaela Verucchi a0e4e9f209 Modify yolov4_berkeley link for download, clean to merge with master 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-01 15:17:53 +02:00
Francesco Gatti 18794d52c2 darknet parser tested 2020-06-01 14:55:08 +02:00