diff --git a/modules/__init__.py b/modules/__init__.py index 7968738..45a9cac 100644 --- a/modules/__init__.py +++ b/modules/__init__.py @@ -12,7 +12,7 @@ def imwrite_unicode(path, img, params=None): if not ext: ext = ".png" result, encoded_img = cv2.imencode(ext, img, params if params else []) - if result: + result, encoded_img = cv2.imencode(f".{ext}", img, params if params is not None else []) encoded_img.tofile(path) return True return False \ No newline at end of file