diff --git a/photo_mask/photo_mask.py b/photo_mask/photo_mask.py index 1ed6dec..ecbbe90 100644 --- a/photo_mask/photo_mask.py +++ b/photo_mask/photo_mask.py @@ -126,6 +126,7 @@ def get_mask_layout(image, name, id_card_num): r = f"[{']['.join(re_list)}]" if (name_len & 1) == 1: r += f"|[{']['.join(re_list[:name_offset])}].?[{']['.join(re_list[-name_offset:])}]" + r += f"|名[:|:](?=.*[{'|'.join(re_list)}]).{{{name_len}}}" for layout in layouts: find_name_by_key = True