Fix bug where expected color chanel order was wrong

obss · Nov 6, 2023 · ae67e75 · ae67e75
1 parent 3a21a06
commit ae67e75
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/sahi/models/onnx_model.py b/sahi/models/onnx_model.py
@@ -76,10 +76,9 @@ def _preprocess_image(self, image: np.ndarray, input_shape: Tuple[int, int]) ->
 
         Args:
             image: np.ndarray
-                Input image with color channel order BGR.
+                Input image with color channel order RGB.
         """
         input_image = cv2.resize(image, input_shape)
-        input_image = cv2.cvtColor(input_image, cv2.COLOR_BGR2RGB)
 
         input_image = input_image / 255.0
         input_image = input_image.transpose(2, 0, 1)