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

286 Commits

Author SHA1 Message Date
Davide Sapienza 59b0f434a7 Fix a bug in the 3D bounding boxes. 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-12-14 15:32:35 +01:00
Davide Sapienza dbc052865c Fix a wrong path in CenternetDetection3DTrack.cpp 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-12-09 18:04:42 +01:00
Davide Sapienza 48ecebe6dd Add CenterTrack pre, post, visualization and demo. 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-12-07 18:45:29 +01:00
Davide Sapienza 7f65ee0b2a Merge remote-tracking branch 'origin/master' into cnet 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-12-07 17:45:44 +01:00
Micaela Verucchi a52e18b6e6 Add shelfnet_mapillary, README_seg, resize of input 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-11-24 12:37:33 +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 702791e41a Add support for yolov4x-mish. 
Changes:
- add parameters nms_kind, nms_thresh, new_coords to yolo layer and darknet parser
- added diou nms, new method to compute the BBs
- created test for yolov4x-mish called yolo4x

Tested, all tests work.
Problem to solve: little loss in mAP of yolo4x

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-11-23 11:29:32 +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 df5443e017 Fix boxes also for Centernet 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-08-06 15:53:57 +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 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
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 a68f45cb4e Merge with master, add yolov4tiny_512 
Signed-off-by: tk <micaelaverucchi@gmail.com>
 2020-07-06 17:57:50 +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
Micaela Verucchi d25803d438 Post-processing improved 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-24 18:46:25 +02:00
Micaela Verucchi 082920f3f5 Shelfnet works, also visualization. Postprocessing need to be parallelized 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-23 20:01:47 +02:00
Micaela Verucchi 94e558003d Shelfnet works on tensorRT (shortcut need to be fixed) 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-23 12:50:24 +02:00
Micaela Verucchi 6fd261f628 Shelfnet works on cuDNN. To test everything else 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-22 18:11:19 +02:00
Omar Alvarez 2817ade782 Fix parsing label files with unexpected chars 2020-06-22 13:40:15 +02:00
Micaela Verucchi 9f1e30eaa9 Add shelfnet. Resnet18backbone works 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-19 18:55:42 +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
Francesco Gatti cbfc8ea4f2 serialize fix 2020-06-14 13:01:45 +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
Francesco Gatti 20303ac32e cudnn8 compile 2020-06-09 20:53:05 +02:00
Francesco Gatti 62e4a3f779 memory release 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 d4e0d07e09 merge 2020-06-01 15:34:59 +02:00
Francesco Gatti 18794d52c2 darknet parser tested 2020-06-01 14:55:08 +02:00
Francesco Gatti d2e2669b6d darknet parse all net to be tested 2020-06-01 12:22:55 +02:00
Micaela Verucchi d936e5f740 Add mish_yashas 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-30 16:57:53 +02:00
Davide Sapienza ba8c282384 Add 3D CenterNet detection class 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-05-27 18:03:09 +02:00
Micaela Verucchi 854e1d909a Fix original size 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-19 16:33:45 +02:00
Micaela Verucchi 5d01a3f629 Fix Centernet postprocessing 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-14 17:13:38 +02:00
Micaela Verucchi 40456592fc Adapt detection classes to use batches, adapt demos, update README 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-14 16:41:51 +02:00
Micaela Verucchi f684a2126e Merge branch 'master' of https://github.com/ceccocats/tkDNN into eval 2020-05-11 11:58:51 +02:00
Micaela Verucchi 533bb48789 Add json detection creation for codalab check 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-11 11:57:58 +02:00
Francesco Gatti 85bbbf42a1 merge 2020-05-03 15:58:00 +02:00
Micaela Verucchi db0f8a4d99 Support yolov4 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-28 20:35:30 +02:00
Francesco Gatti e6d435a463 Merge commit '04bd5d7ff46270c732b76cd9656e22de930e138e' into tree 2020-04-28 14:54:18 +02:00
Micaela Verucchi a874fad2bd Merge branch 'master' of https://github.com/ceccocats/tkDNN into cnet 2020-04-28 14:43:24 +02:00
dsapienza 6e5ab031c4 Deformable batch works 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-04-25 02:47:31 +02:00
Micaela Verucchi f418a1f9b2 Add bdd new tests, add BDD100K_val download, fix yolo3-512 link 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-24 12:39:57 +02:00
Francesco Gatti 3ba276a236 merge tkDNN 2020-04-24 11:18:22 +02:00
Francesco Gatti 3d940a9fa2 batch seems ok in yolo3_berkely 
layers to be checked:
DeformableConvRT
FlattenConcatRT
ReshapeRT
RouteRT (dont know why but seems working)
 2020-04-21 19:41:40 +00:00