diff --git a/photo_mask.py b/photo_mask.py index 6e04ed9..e28f68e 100644 --- a/photo_mask.py +++ b/photo_mask.py @@ -241,13 +241,18 @@ def photo_mask(pk_phhd, content): split_result = split_image(image) for img in split_result: angles = get_image_rotation_angles(img["img"]) - rotated_img = rotate_image(img["img"], int(angles[0])) + angle = int(angles[0]) + rotated_img = rotate_image(img["img"], angle) results = get_mask_layout(rotated_img, content) + if not results: + angle = int(angles[1]) + rotated_img = rotate_image(img["img"], angle) + results = get_mask_layout(rotated_img, content) for result in results: height, width = img["img"].shape[:2] center = (width / 2, height / 2) - result = rotate_rectangle(result, center, int(angles[0])) + result = rotate_rectangle(result, center, angle) result = ( result[0] + img["x_offset"], result[1] + img["y_offset"],