增加图片损坏的判断

2025-06-26 08:48:11 +08:00
parent cd604bc1eb
commit a67c53f470
2 changed files with 3 additions and 1 deletions
--- a/photo_mask/auto_photo_mask.py
+++ b/photo_mask/auto_photo_mask.py
@@ -199,6 +199,8 @@ def mask_photo(img_url, name, id_card_num, color=(255, 255, 255)):

    # 打开图片
    image = image_util.read(img_url)
+    if not image:
+        return False, image
    original_image = image
    is_masked, image = _mask(image, name, id_card_num, color)
    if not is_masked: