convert color from bgr to rgb

zhiqwang · Apr 21, 2021 · 9ed79f2 · 9ed79f2
1 parent 11e6612
commit 9ed79f2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/yolort/utils/image_utils.py b/yolort/utils/image_utils.py
@@ -175,6 +175,7 @@ def read_image_to_tensor(
         image (np.ndarray): the candidate ndarray image to be parsed to Tensor.
         is_half (bool): whether to transfer image to half. Default: False.
     """
+    image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
     image = np.ascontiguousarray(image, dtype=np.float32)  # uint8 to float32
     image = np.transpose(image / 255.0, [2, 0, 1])