Merge pull request meituan#780 from meituan/fix_preprocess_bug_in_hub…

…conf

Fix preprocess bug in hubconf

hubconf.py

@@ -1,4 +1,5 @@
 import os
+import cv2
 import math
 import pathlib
 import torch
@@ -70,9 +71,16 @@ def make_divisible(x, divisor):
 
 def process_image(path, img_size, stride):
     '''Preprocess image before inference.'''
-    img_src = np.asarray(Image.open(path).convert('RGB'))
+    try:
+            img_src = cv2.imread(path)
+            img_src = cv2.cvtColor(img_src, cv2.COLOR_RGB2BGR)
+            assert img_src is not None, f"opencv cannot read image correctly or {path} not exists"
+    except:
+            img_src = np.asarray(Image.open(path))
+            assert img_src is not None, f"Image Not Found {path}, workdir: {os.getcwd()}"
+
     image = letterbox(img_src, img_size, stride=stride)[0]
-    image = image.transpose((2, 0, 1))[::-1]  # HWC to CHW, BGR to RGB
+    image = image.transpose((2, 0, 1)) # HWC to CHW
     image = torch.from_numpy(np.ascontiguousarray(image))
     image = image.float()
     image /= 255