从分割后就判断图片是否为空

2024-08-28 15:25:41 +08:00
parent e5b9ff69a6
commit 06b637544a
2 changed files with 3 additions and 1 deletions
--- a/photo_review/auto_photo_review.py
+++ b/photo_review/auto_photo_review.py
@@ -92,6 +92,8 @@ def information_extraction(ie, phrecs, identity):
            zx_ie_results = []
            split_results = image_util.split(dewarped_image)
            for split_result in split_results:
+                if split_result["img"] is None or split_result["img"].size == 0:
+                    continue
                rotated_img = image_util.rotate(split_result["img"], int(angles[0]))
                ie_results = [{"result": ie_temp_image(ie, OCR, rotated_img), "angle": angles[0]}]
                if not ie_results[0]["result"] or len(ie_results[0]["result"]) < len(ie.kwargs.get("schema")):