Commit Graph

29 Commits

Author SHA1 Message Date
Francesco Gatti df31375b67 Merge branch 'master' into tensorrt8 2022-03-30 16:41:45 +02:00
Micaela Verucchi 04de9908a6 Add yolo4-csp for crowdhuman dataset, add shelfnet for coco-stuff dataset, fix minor in demo.cpp
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2021-11-29 17:33:57 +01:00
Micaela Verucchi eca10ac0a8 Update weights
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2021-09-19 00:16:31 +02:00
perseusdg 03473743c4 tkDNN works with trt8!!,need to test int8 and mobilenet,dla_cnet (fps seems to be a bit low 350 on trt8 compared to 396 on trt7) 2021-09-07 22:00:28 +05:30
perseusdg 504dee5016 test_yolo4 netRT->destroy() 2021-09-06 15:35:57 +05:30
perseusdg de83ae5d25 update tensorrt8 branch 2021-08-30 19:04:26 +05:30
Micaela Verucchi f4e71a5a28 Add yolo4_berkeley_f1
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2021-07-21 12:56:00 +02:00
Micaela Verucchi 89b1bb7bee Add missing files
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2021-07-20 18:39:25 +02:00
Micaela Verucchi 318dfe4b55 merge with master
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2021-07-20 18:38:46 +02:00
Francesco Gatti 6611a91201 merge perseusdg windows10 support 2021-04-24 18:55:26 +02:00
hchandirasekar f12ec3c935 added yolo4x and yolo4-csp from the github repo and download file corrections 2021-03-26 12:18:46 +05:30
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 adac8576b0 Add support to Scaled-YOLO4, update Yolov4x-mish (tested)
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2021-01-22 17:57:42 +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 86478f9384 Add yolo4_mmr test
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-10-23 11:40:55 +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 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 1dfc69ba89 viz yolo3 preprocess 2020-06-16 13:19:30 +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
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 c4e955eab5 Add different yolov4 size tests
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-06-11 12:43:34 +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