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-25 16:05:18 +08:00
parent 3538e9d19d
commit 353e8572ef
1 changed files with 18 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
photo_mask/photo_mask.py
View File

@@ -13,7 +13,7 @@ from ucloud import BUCKET, ufile
from util import image_util, util from util import image_util, util
def find_boxes(content, layout, offset=0, length=None, improve=False, image_path=None): def find_boxes(content, layout, offset=0, length=None, improve=False, image_path=None, extra_content=None):
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]
@@ -67,8 +67,13 @@ def find_boxes(content, layout, offset=0, length=None, improve=False, image_path
except Exception as e: except Exception as e:
# 如果出现其他错误，抛出 # 如果出现其他错误，抛出
raise e raise e
if extra_content:
contents = extra_content.append(content)
else:
contents = [content]
for layout in layouts: for layout in layouts:
if content in layout[1]: for c in contents:
if c in layout[1]:
temp_box = find_boxes(content, layout)[0] temp_box = find_boxes(content, layout)[0]
if temp_box: if temp_box:
boxes.append([ boxes.append([
@@ -77,6 +82,7 @@ def find_boxes(content, layout, offset=0, length=None, improve=False, image_path
temp_box[2] + capture_box[0] - offset_x, temp_box[2] + capture_box[0] - offset_x,
temp_box[3] + capture_box[1] - offset_y, temp_box[3] + capture_box[1] - offset_y,
]) ])
break
util.delete_temp_file(temp_file.name) util.delete_temp_file(temp_file.name)
if not boxes: if not boxes:
@@ -130,7 +136,8 @@ def get_mask_layout(image, name, id_card_num):
matches = re.findall(r, layout[1]) matches = re.findall(r, layout[1])
for match in matches: for match in matches:
if match in layout[1]: if match in layout[1]:
result += find_boxes(match, layout, improve=True, image_path=temp_file.name) result += find_boxes(match, layout, improve=True, image_path=temp_file.name,
extra_content=name_list)
find_name_by_key = False find_name_by_key = False
break break
if id_card_num in layout[1]: if id_card_num in layout[1]: