Quellcode durchsuchen

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

gaotingquan vor 3 Wochen
Ursprung
Commit
c1ca6604ea
1 geänderte Dateien mit 3 neuen und 1 gelöschten Zeilen
  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):