diff --git a/sahi/models/onnx_model.py b/sahi/models/onnx_model.py index a4bb61f06..001a48b71 100644 --- 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)