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>
This commit is contained in:
Micaela Verucchi
2020-11-23 11:25:52 +01:00
parent 86478f9384
commit 702791e41a
10 changed files with 1571 additions and 31 deletions
+1
View File
@@ -73,6 +73,7 @@ do
print_output $? imuodom
test_net yolo4
test_net yolo4x
test_net yolo4_berkeley
test_net yolo4tiny
test_net yolo3