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 14:45:08 +08:00
parent faf3c33733
commit d0407d568e
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
photo_mask/photo_mask.py
View File

@@ -113,6 +113,11 @@ def get_mask_layout(image, name, id_card_num):
for sc in SIMILAR_CHAR[char]: for sc in SIMILAR_CHAR[char]:
name_list.append(name.replace(char, sc)) name_list.append(name.replace(char, sc))
name_len = len(name)
name_offset = int(name_len / 2)
r = f"{name[:-1]}.?|.?{name[1:]}" + (
f"|{name[:name_offset]}.{name[-name_offset:]}" if (name_len & 1) == 1 else "")
for layout in layouts: for layout in layouts:
find_name_by_key = True find_name_by_key = True
find_id_card_num_by_key = True find_id_card_num_by_key = True
@@ -121,11 +126,7 @@ def get_mask_layout(image, name, id_card_num):
result += find_boxes(name, layout, improve=True, image_path=temp_file.name) result += find_boxes(name, layout, improve=True, image_path=temp_file.name)
find_name_by_key = False find_name_by_key = False
break break
if find_name_by_key and len(name) > 2: if find_name_by_key and name_len > 2:
name_len = len(name)
name_offset = int(name_len / 2)
r = f"{name[:-1]}.?|.?{name[1:]}" + (
f"|{name[:name_offset]}.{name[-name_offset:]}" if (name_len & 1) == 1 else "")
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]: