From f44080be834093a2fa032259045cf1caaccdd213 Mon Sep 17 00:00:00 2001
From: Michael Heilig <michael.heilig@querplaner.de>
Date: Thu, 27 Jul 2023 22:55:18 +0200
Subject: [PATCH] added parameter "agnostic_nms=True"

---
 OTVision/detect/yolo.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/OTVision/detect/yolo.py b/OTVision/detect/yolo.py
index 9564c8d5..1615130b 100644
--- 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]: