瀏覽代碼

refactor imwrite_cv, now return a boolean value

quarrying 2 年之前
父節點
當前提交
1821bf261e
共有 1 個文件被更改,包括 6 次插入2 次删除
  1. 6 2
      khandy/image/misc.py

+ 6 - 2
khandy/image/misc.py

@@ -53,8 +53,12 @@ def imread_cv(file_or_buffer, flags=-1):
 def imwrite_cv(filename, image, params=None):
 def imwrite_cv(filename, image, params=None):
     """Improvement on cv2.imwrite, make it support filename including chinese character.
     """Improvement on cv2.imwrite, make it support filename including chinese character.
     """
     """
-    cv2.imencode(os.path.splitext(filename)[-1], image, params)[1].tofile(filename)
-    
+    try:
+        cv2.imencode(os.path.splitext(filename)[-1], image, params)[1].tofile(filename)
+        return True
+    except:
+        return False
+
 
 
 def imwrite_bytes(filename, image_bytes, update_extension=True):
 def imwrite_bytes(filename, image_bytes, update_extension=True):
     extension = imghdr.what('', image_bytes)
     extension = imghdr.what('', image_bytes)