Update general.py

nikitabelooussovbtis · Oct 13, 2022 · bc21628 · bc21628
1 parent 8dfa3cf
commit bc21628
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/general.py b/utils/general.py
@@ -778,7 +778,7 @@ def non_max_suppression(prediction,
                         iou_thres=0.45,
                         classes=None,
                         agnostic=False,
-                        multi_label=False,
+                        multi_label=True,
                         labels=(),
                         max_det=300):
     """Non-Maximum Suppression (NMS) on inference results to reject overlapping bounding boxes