Эх сурвалжийг харах

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

gaotingquan 3 долоо хоног өмнө
parent
commit
c1ca6604ea

+ 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):