diff --git a/photo_mask/photo_mask.py b/photo_mask/photo_mask.py index b6639ca..fe3fae9 100644 --- a/photo_mask/photo_mask.py +++ b/photo_mask/photo_mask.py @@ -119,6 +119,7 @@ def get_mask_layout(image, name, id_card_num): if name in layout[1]: result += find_boxes(name, layout, improve=True, image_path=temp_file.name) find_name_by_key = False + break if id_card_num in layout[1]: result += find_boxes(id_card_num, layout, improve=True, image_path=temp_file.name) find_id_card_num_by_key = False