added parameter "agnostic_nms=True"

OpenTrafficCam · Jul 27, 2023 · f44080b · f44080b
1 parent 593ac8f
commit f44080b
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]: