Try faster threading

OpenTrafficCam · Sep 24, 2024 · ec97915 · ec97915
1 parent 5010b8f
commit ec97915
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/OTVision/detect/yolo.py b/OTVision/detect/yolo.py
@@ -147,6 +147,7 @@ def _load_model(self) -> YOLOv8:
 
     def _predict(self, video: Path) -> Generator[Results, None, None]:
         with av.open(str(video.absolute())) as container:
+            container.streams.video[0].thread_type = "AUTO"
             for frame in container.decode(video=0):
                 results = self.model.predict(
                     source=frame.to_ndarray(format="rgb24"),