Commit Graph

208 Commits

Author SHA1 Message Date
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
Francesco Gatti 6a185564e7 RegionRT softmax fix 2020-04-24 00:08:06 +02:00
Francesco Gatti afaeb75d84 Reshape batch fix 2020-04-23 23:51:23 +02:00
Francesco Gatti 2a6f7db4af Flatten batch to be checked 2020-04-23 01:18:27 +02:00