From 9ecd2dab8311286e47e205a25ccd526f75d7564b Mon Sep 17 00:00:00 2001 From: KUNJ SHAH Date: Mon, 5 May 2025 13:10:00 +0000 Subject: [PATCH] changes --- modules/__init__.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/modules/__init__.py b/modules/__init__.py index 8b50aa7..c740bae 100644 --- a/modules/__init__.py +++ b/modules/__init__.py @@ -1,3 +1,4 @@ +import os import cv2 import numpy as np @@ -11,8 +12,11 @@ cv2.imread = imread_unicode # Define a function to support unicode characters in file paths when saving def imwrite_unicode(path, img, params=None): # Encode the image - ext = path.split(".")[-1] # Get file extension - result, encoded_img = cv2.imencode(f".{ext}", img, params if params else []) + root, ext = os.path.splitext(path) + # If no extension is found, you can choose a default extension (e.g., ".png") + if not ext: + ext = ".png" + result, encoded_img = cv2.imencode(ext, img, params if params else []) if result: encoded_img.tofile(path) # Save image using numpy's `tofile()` @@ -20,4 +24,4 @@ def imwrite_unicode(path, img, params=None): return False # Override `cv2.imwrite` -cv2.imwrite = imwrite_unicode +cv2.imwrite = imwrite_unicode \ No newline at end of file