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
Francesco Gatti
d2e2669b6d
darknet parse all net to be tested
2020-06-01 12:22:55 +02:00
Micaela Verucchi
db0f8a4d99
Support yolov4
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2020-04-28 20:35:30 +02:00
fbagni
7e7f480e2b
Yolov3_tiny works on tensorRT
...
Signed-off-by: fbagni <gattinomicino>
2019-12-23 16:11:41 +01:00
nvidia
854a4c316a
Add ResizeLayerRT plugin
...
Signed-off-by: nvidia <micaelaverucchi@gmail.com >
2019-12-20 15:27:41 +01:00
Micaela Verucchi
44b2bce3ff
Yolo3_tiny CUDNN works, TensorRT doesn't. Add n_masks to Yolo layer.
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2019-12-05 20:43:48 +00:00
Micaela Verucchi
a2db98670a
Add Yolov3 (COCO80) and Yolov3-tiny (COCO80), TensorRT for tiny not working
...
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com >
2019-12-04 17:01:40 +00:00
Francesco Gatti
ec02c7292f
save layer names in rt file
2019-09-16 19:41:59 +02:00
mbosi
8c629ebe7b
string input and flir test
2019-09-14 19:03:13 +02:00
Francesco Gatti
c7941666ec
demo for more yolo3
2019-02-19 08:43:35 +00:00
Francesco Gatti
87fe342ca2
yoloRT load anchors
2019-02-18 21:39:14 +01:00
Francesco Gatti
2c63bf05be
multipl yolo morge
2019-02-06 22:24:01 +00:00
Francesco Gatti
0e97452460
dects dont works
2019-02-05 20:09:47 +00:00
Francesco Gatti
c8dea4668d
compute detections
2019-02-04 20:34:15 +00:00
Francesco Gatti
7a51b4382d
yolo3 ok
2018-12-21 15:28:47 +01:00
Francesco Gatti
2ab47b5874
yolo layer
2018-12-20 16:10:01 +01:00