diff --git a/photo_review/auto_photo_review.py b/photo_review/auto_photo_review.py index a9b1317..7596e57 100644 --- 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")): diff --git a/util/image_util.py b/util/image_util.py index ec80cdf..0b30cc0 100644 --- a/util/image_util.py +++ b/util/image_util.py @@ -111,7 +111,7 @@ def rotate(image, angle): :param angle: 逆时针旋转角度 :return: 旋转后的图片NumPy数组 """ - if angle == 0 or image is None or image.size == 0: + if angle == 0: return image height, width = image.shape[:2] if angle == 180: