The nms type is non-agnostic actually

zhiqwang · Jan 23, 2022 · 43d4f6e · 43d4f6e
1 parent eed04f4
commit 43d4f6e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_utils.py b/test/test_utils.py
@@ -75,7 +75,7 @@ def test_load_from_ultralytics_voc(arch: str, version: str, upstream_version: st
     model_yolov5.eval()
     with torch.no_grad():
         outs = model_yolov5(img[None])[0]
-        outs = non_max_suppression(outs, conf, iou, agnostic=True)
+        outs = non_max_suppression(outs, conf, iou)
         out_from_yolov5 = outs[0]
 
     # Define yolort model