pull/1210/head
KUNJ SHAH 2025-05-05 13:19:29 +00:00
parent 9ecd2dab83
commit fe4a87e8f2
1 changed files with 4 additions and 13 deletions

View File

@ -2,26 +2,17 @@ import os
import cv2 import cv2
import numpy as np import numpy as np
# Define a new function that supports unicode characters in file paths # Utility function to support unicode characters in file paths for reading
def imread_unicode(path, flags=cv2.IMREAD_COLOR): def imread_unicode(path, flags=cv2.IMREAD_COLOR):
return cv2.imdecode(np.fromfile(path, dtype=np.uint8), flags) return cv2.imdecode(np.fromfile(path, dtype=np.uint8), flags)
# Override the original `cv2.imread` # Utility function to support unicode characters in file paths for writing
cv2.imread = imread_unicode
# Define a function to support unicode characters in file paths when saving
def imwrite_unicode(path, img, params=None): def imwrite_unicode(path, img, params=None):
# Encode the image
root, ext = os.path.splitext(path) root, ext = os.path.splitext(path)
# If no extension is found, you can choose a default extension (e.g., ".png")
if not ext: if not ext:
ext = ".png" ext = ".png"
result, encoded_img = cv2.imencode(ext, img, params if params else []) result, encoded_img = cv2.imencode(ext, img, params if params else [])
if result: if result:
encoded_img.tofile(path) # Save image using numpy's `tofile()` encoded_img.tofile(path)
return True return True
return False return False
# Override `cv2.imwrite`
cv2.imwrite = imwrite_unicode