diff --git a/photo_mask.py b/photo_mask.py index 112bf72..c696855 100644 --- a/photo_mask.py +++ b/photo_mask.py @@ -247,6 +247,9 @@ def photo_mask(pk_phhd, content): angle = int(angles[1]) rotated_img = rotate_image(img["img"], angle) results = get_mask_layout(rotated_img, content) + if not results and "0" not in angles: + angle = 0 + results = get_mask_layout(img["img"], content) if results: is_masked = True