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-11 18:09:38 +08:00
parent fee59b3133
commit e1beedec96
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
photo_mask.py
View File

@@ -196,7 +196,7 @@ def zoom_box(box, ratio):
return [x1, y1, x2, y2] return [x1, y1, x2, y2]
def find_box_of_content(content, layout, img_path): def find_box_of_content(content, layout, img_path, improve=True):
full_box = layout[0] full_box = layout[0]
x_len = full_box[2] - full_box[0] x_len = full_box[2] - full_box[0]
y_len = full_box[3] - full_box[1] y_len = full_box[3] - full_box[1]
@@ -232,7 +232,7 @@ def find_box_of_content(content, layout, img_path):
] ]
is_abnormal = box[3] - box[1] < (box[2] - box[0]) * len(content) / 2 is_abnormal = box[3] - box[1] < (box[2] - box[0]) * len(content) / 2
if is_abnormal: if is_abnormal and improve:
# 比例异常，再次识别 # 比例异常，再次识别
image = cv2.imread(img_path) image = cv2.imread(img_path)
# 截图时偏大一点 # 截图时偏大一点
@@ -250,7 +250,7 @@ def find_box_of_content(content, layout, img_path):
raise e raise e
for layout in layouts: for layout in layouts:
if content in layout[1]: if content in layout[1]:
temp_box = find_box_of_content(content, layout, temp_file.name) temp_box = find_box_of_content(content, layout, temp_file.name, False)
if temp_box: if temp_box:
box = [ box = [
temp_box[0] + capture_box[0], temp_box[0] + capture_box[0],