multi_label always true for val_onnx (#117)

neuralmagic · Oct 20, 2022 · 6bf3a87 · 6bf3a87
1 parent 656953a
commit 6bf3a87
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/val_onnx.py b/val_onnx.py
@@ -310,7 +310,10 @@ def run(
 
         # Inference
         out = yolo_pipeline(
-            images=[im.numpy()], iou_thres=iou_thres, conf_thres=conf_thres
+            images=[im.numpy()], 
+            iou_thres=iou_thres, 
+            conf_thres=conf_thres, 
+            multi_label=True
         )
 
         # inference, loss outputs