Convert frames to RGB before SAHI inference (#17795)

Co-authored-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 months ago · 54d845baea
parent de89616212
commit 54d845baea
1 changed files with 1 additions and 1 deletions
--- a/examples/YOLOv8-SAHI-Inference-Video/yolov8_sahi.py
+++ b/examples/YOLOv8-SAHI-Inference-Video/yolov8_sahi.py
@ -64,7 +64,7 @@ class SAHIInference:
                break
            annotator = Annotator(frame)  # Initialize annotator for plotting detection and tracking results
            results = get_sliced_prediction(
-                frame,
+                frame[..., ::-1],
                self.detection_model,
                slice_height=512,
                slice_width=512,