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

211 Commits

Author SHA1 Message Date
perseusdg b994eb2a4e 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 936068e9b4 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 a194ba59f6 Fix typos (#107) 
Signed-off-by: micaela <micaelaverucchi@gmail.com>
 2020-09-11 09:13:59 +02:00
Micaela Verucchi 498b163f36 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 5c7f1c283e Patch for CUDNN 8.0.1 
Signed-off-by: tk <micaelaverucchi@gmail.com>
 2020-07-16 18:16:09 +02:00
Francesco Gatti 8fad424c61 Yolo4-tiny batched 
fix #59
 2020-06-30 19:37:16 +02:00
Micaela Verucchi 4c9b9bf03a yolo4tiny works on tensorRT 🐬 🐬 🐬 🐬 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-30 15:52:58 +02:00
Micaela Verucchi 35c0d9fd37 yolov4tiny works on CUDNN 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-30 15:19:03 +02:00
Francesco Gatti 517eaa553b dealloc in test.h 
fix #36
 2020-06-16 12:44:12 +02:00
Francesco Gatti f9c50b2a49 darknet parser cpp 2020-06-16 12:41:48 +02:00
Francesco Gatti 0f3dbac962 viz yolo3 2020-06-16 12:36:45 +02:00
Francesco Gatti 4f907dde40 memory release 2020-06-02 12:43:06 +02:00
Francesco Gatti 8c79b75227 release layer wgs and version update 2020-06-01 21:03:37 +02:00
Francesco Gatti 235a6838eb all test ok 2020-06-01 16:15:29 +02:00
Francesco Gatti 2df8bb92c2 merge 2020-06-01 15:34:59 +02:00
Micaela Verucchi fd00822a3e 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 af9bb411be darknet parser tested 2020-06-01 14:55:08 +02:00
Francesco Gatti 21518c5eef darknet parse all net to be tested 2020-06-01 12:22:55 +02:00
Micaela Verucchi 9ee8c5eb58 Check 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-01 10:39:13 +02:00
Micaela Verucchi fe07c8fb26 Read classes' names from file 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-06-01 10:34:48 +02:00
Francesco Gatti 4b7e80ae0d yolo3 parsed ok 2020-05-30 22:11:00 +02:00
Francesco Gatti 99f126320b Merge branch 'darknetparser' of https://github.com/ceccocats/tkDNN into darknetparser 2020-05-30 19:38:31 +02:00
Francesco Gatti 27a83e79d3 darknet parser to be tested on yolo3 2020-05-30 19:38:26 +02:00
Micaela Verucchi 93ffbcf8b2 Add group field 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-30 19:03:29 +02:00
Micaela Verucchi 3e701e7ba3 Add fields to darknetParseFields 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-30 18:44:17 +02:00
Francesco Gatti 664a717353 parse yolo layers 2020-05-30 18:44:02 +02:00
Micaela Verucchi 15e4ab4436 Modify darknetFields_t 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-30 18:10:14 +02:00
Francesco Gatti 3951ff2582 layer parser 2020-05-30 18:02:20 +02:00
Micaela Verucchi 26e6ba52a5 Add some fields 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-30 18:00:22 +02:00
Francesco Gatti 1755e734b1 parse layer and network 2020-05-30 17:59:00 +02:00
Micaela Verucchi 98173ec15c Add darknetParseFields 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-30 17:57:48 +02:00
Micaela Verucchi 8b037781f4 merge 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-30 17:33:18 +02:00
Francesco Gatti 179e07e830 parse line by line 2020-05-30 17:27:22 +02:00
Micaela Verucchi cb6dc4f182 Add parseType 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-30 17:25:48 +02:00
Francesco Gatti f9d28c58ca darknet parser interface 2020-05-30 16:58:40 +02:00
Micaela Verucchi 9618d5fc7f Fix original size 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-19 16:33:45 +02:00
Micaela Verucchi f2991bec5c Add variable batch for detector update 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-19 09:32:27 +02:00
Micaela Verucchi 360adbd1a1 Add verbose define 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-05-15 09:41:27 +02:00
Micaela Verucchi 7df266cac4 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 b8686186e4 Merge branch 'master' of https://github.com/ceccocats/tkDNN into eval 2020-05-11 11:58:51 +02:00
Micaela Verucchi 5eb0cbfcef 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 54074780cf filter yaw ok 2020-05-03 19:12:10 +02:00
Francesco Gatti 3fcb05be64 Merge branch 'tree' into ipmslam 2020-05-03 15:58:43 +02:00
Francesco Gatti 8d45259b15 SEP model 2020-05-03 15:54:39 +02:00
Francesco Gatti bdace759a9 merge tree 2020-05-01 15:39:00 +02:00
Micaela Verucchi 1f4f27d752 Support yolov4 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
 2020-04-28 20:35:30 +02:00
Francesco Gatti f91fb7f287 stream 2020-04-28 14:36:02 +02:00
Micaela Verucchi dbe54f4b67 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 3aa3097bbb sensor close if not started fix 2020-04-28 12:59:30 +02:00
dsapienza 26d8d009c5 Deformable batch works 
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
 2020-04-25 02:47:31 +02:00