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

211 Commits

Author SHA1 Message Date
perseusdg 56feb54377 able to build kernels as shared object file(dll),and minor changes to lstm.cpp and utils.cpp to overcome minor msvc build errors 2021-01-21 00:22:15 +04: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 fa2b3d26cb Fix typos (#107) 
Signed-off-by: micaela <micaelaverucchi@gmail.com>
 2020-09-11 09:13:59 +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
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 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 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
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 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 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
Francesco Gatti d2e2669b6d darknet parse all net to be tested 2020-06-01 12:22:55 +02:00
Micaela Verucchi a6eef498fa Check 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-01 10:39:13 +02:00
Micaela Verucchi 826fcc97c8 Read classes' names from file 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-01 10:34:48 +02:00
Francesco Gatti 6d81473b2a yolo3 parsed ok 2020-05-30 22:11:00 +02:00
Francesco Gatti 0621ed223b Merge branch 'darknetparser' of https://github.com/ceccocats/tkDNN into darknetparser 2020-05-30 19:38:31 +02:00
Francesco Gatti 0ae96a6bc4 darknet parser to be tested on yolo3 2020-05-30 19:38:26 +02:00
Micaela Verucchi d45947eec5 Add group field 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-30 19:03:29 +02:00
Micaela Verucchi e5e6654b1d Add fields to darknetParseFields 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-30 18:44:17 +02:00
Francesco Gatti 298487d5c5 parse yolo layers 2020-05-30 18:44:02 +02:00
Micaela Verucchi 6b6dbabba3 Modify darknetFields_t 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-30 18:10:14 +02:00
Francesco Gatti e1d5f58c3f layer parser 2020-05-30 18:02:20 +02:00
Micaela Verucchi 041a5cf65d Add some fields 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-30 18:00:22 +02:00
Francesco Gatti 8935d85e91 parse layer and network 2020-05-30 17:59:00 +02:00
Micaela Verucchi 64d22c51f1 Add darknetParseFields 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-30 17:57:48 +02:00
Micaela Verucchi eec8de3efa merge 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-30 17:33:18 +02:00
Francesco Gatti 548a3dd33c parse line by line 2020-05-30 17:27:22 +02:00
Micaela Verucchi 15105e90d3 Add parseType 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-30 17:25:48 +02:00
Francesco Gatti 4e1c7a70b1 darknet parser interface 2020-05-30 16:58:40 +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 377310af50 Add variable batch for detector update 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-19 09:32:27 +02:00
Micaela Verucchi 23b40de508 Add verbose define 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-15 09:41:27 +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 ca7631250c filter yaw ok 2020-05-03 19:12:10 +02:00
Francesco Gatti 7f0d9a930d Merge branch 'tree' into ipmslam 2020-05-03 15:58:43 +02:00
Francesco Gatti 5a9ed44b6a SEP model 2020-05-03 15:54:39 +02:00
Francesco Gatti 98537624cf merge tree 2020-05-01 15:39: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 04bd5d7ff4 stream 2020-04-28 14:36:02 +02:00
Micaela Verucchi 89f91f568e Fix shortcutRT plugin, now works with batches, fix yolo3_512 downlaod link 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-28 14:34:12 +02:00
Francesco Gatti c975a467b1 sensor close if not started fix 2020-04-28 12:59:30 +02:00
dsapienza 6e5ab031c4 Deformable batch works 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-04-25 02:47:31 +02:00