Merge NMS update

2020-03-26 12:48:00 -07:00 · 2020-03-26 12:48:00 -07:00 · a322fc5d4b
parent 171b4129b5
commit a322fc5d4b
1 changed files with 1 additions and 0 deletions
--- a/utils/utils.py
+++ b/utils/utils.py
@ -564,6 +564,7 @@ def non_max_suppression(prediction, conf_thres=0.1, iou_thres=0.6, multi_label=T
            i = iou.max(0)[0] < iou_thres

        output[xi] = x[i]
+    return output


 def get_yolo_layers(model):