浏览代码

fix: use cv2.imdecode to support reading files with Chinese characters in filename

gaotingquan 3 周之前
父节点
当前提交
c1ca6604ea
共有 1 个文件被更改,包括 3 次插入1 次删除
  1. 3 1
      paddlex/inference/utils/io/readers.py

+ 3 - 1
paddlex/inference/utils/io/readers.py

@@ -267,7 +267,9 @@ class OpenCVImageReaderBackend(_ImageReaderBackend):
 
     def read_file(self, in_path):
         """read image file from path by OpenCV"""
-        return cv2.imread(in_path, flags=self.flags)
+        with open(in_path, "rb") as f:
+            img_array = np.frombuffer(f.read(), np.uint8)
+        return cv2.imdecode(img_array, flags=self.flags)
 
 
 class PILImageReaderBackend(_ImageReaderBackend):