Merge pull request #252 from OpenTrafficCam/bug/2546-nms-is-not-perfo…

…rmed-across-classes Added parameter "agnostic_nms=True" to allow for NMS across classes
OpenTrafficCam · Aug 4, 2023 · e02fead · e02fead
2 parents 593ac8f + f44080b
commit e02fead
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/OTVision/detect/yolo.py b/OTVision/detect/yolo.py
@@ -150,6 +150,7 @@ def _predict(self, video: Path) -> Generator[Results, None, None]:
             stream=True,
             verbose=False,
             batch=-1,
+            agnostic_nms=True
         )
 
     def _parse_detections(self, detection_result: Boxes) -> list[Detection]: