np.fromfile() Chinese image paths fix (#6979)

* 🎉 🆕 now can read Chinese image path. use "cv2.imdecode(np.fromfile(f, np.uint8), cv2.IMREAD_COLOR)" instead of "cv2.imread(f)" for Chinese image path. * Update datasets.py * Update __init__.py Co-authored-by: Glenn Jocher <[email protected]>
ultralytics · Mar 23, 2022 · bc3ed95 · bc3ed95
1 parent c3ae4e4
commit bc3ed95
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/utils/datasets.py b/utils/datasets.py
@@ -32,6 +32,9 @@
                            segments2boxes, xyn2xy, xywh2xyxy, xywhn2xyxy, xyxy2xywhn)
 from utils.torch_utils import torch_distributed_zero_first
 
+# Remap
+cv2.imread = lambda x: cv2.imdecode(np.fromfile(x, np.uint8), cv2.IMREAD_COLOR)  # for Chinese filenames
+
 # Parameters
 HELP_URL = 'https://github.com/ultralytics/yolov5/wiki/Train-Custom-Data'
 IMG_FORMATS = 'bmp', 'dng', 'jpeg', 'jpg', 'mpo', 'png', 'tif', 'tiff', 'webp'  # include image suffixes

diff --git a/utils/loggers/__init__.py b/utils/loggers/__init__.py
@@ -148,6 +148,9 @@ def on_train_end(self, last, best, plots, epoch, results):
 
         if self.tb:
             import cv2
+            import numpy as np
+
+            cv2.imread = lambda x: cv2.imdecode(np.fromfile(x, np.uint8), cv2.IMREAD_COLOR)  # remap for Chinese files
             for f in files:
                 self.tb.add_image(f.stem, cv2.imread(str(f))[..., ::-1], epoch, dataformats='HWC')