Echo/fcb_photo_review
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

添加多角度识别

Browse Source
This commit is contained in:
liuyebo
2024-07-10 10:52:45 +08:00
parent 33db105518
commit 0a7e5b42e8
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
photo_mask.py
View File

@@ -241,13 +241,18 @@ def photo_mask(pk_phhd, content):
split_result = split_image(image) split_result = split_image(image)
for img in split_result: for img in split_result:
angles = get_image_rotation_angles(img["img"]) 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) results = get_mask_layout(rotated_img, content)
for result in results: for result in results:
height, width = img["img"].shape[:2] height, width = img["img"].shape[:2]
center = (width / 2, height / 2) center = (width / 2, height / 2)
result = rotate_rectangle(result, center, int(angles[0])) result = rotate_rectangle(result, center, angle)
result = ( result = (
result[0] + img["x_offset"], result[0] + img["x_offset"],
result[1] + img["y_offset"], result[1] + img["y_offset"],