Pārlūkot izejas kodu

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

gaotingquan 3 nedēļas atpakaļ
vecāks
revīzija
c1ca6604ea
1 mainītis faili ar 3 papildinājumiem un 1 dzēšanām
  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):