From 54d845baea257006cc21025f66881348345be1d1 Mon Sep 17 00:00:00 2001 From: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com> Date: Tue, 26 Nov 2024 18:01:54 +0800 Subject: [PATCH] Convert frames to RGB before SAHI inference (#17795) Co-authored-by: UltralyticsAssistant Co-authored-by: Glenn Jocher --- examples/YOLOv8-SAHI-Inference-Video/yolov8_sahi.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/YOLOv8-SAHI-Inference-Video/yolov8_sahi.py b/examples/YOLOv8-SAHI-Inference-Video/yolov8_sahi.py index 4243cc35b..9e0ba13d9 100644 --- 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,