From d0407d568e135531192bbdfa5bc2fad7a806a22c Mon Sep 17 00:00:00 2001
From: liuyebo <1515783401@qq.com>
Date: Thu, 25 Jul 2024 14:45:08 +0800
Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E6=95=B4=E4=BB=A3=E7=A0=81=E9=A1=BA?=
 =?UTF-8?q?=E5=BA=8F=EF=BC=8C=E5=87=8F=E5=B0=91=E5=8F=98=E9=87=8F=E5=A3=B0?=
 =?UTF-8?q?=E6=98=8E?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 photo_mask/photo_mask.py | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/photo_mask/photo_mask.py b/photo_mask/photo_mask.py
index 40e456f..0352c67 100644
--- a/photo_mask/photo_mask.py
+++ b/photo_mask/photo_mask.py
@@ -113,6 +113,11 @@ def get_mask_layout(image, name, id_card_num):
                     for sc in SIMILAR_CHAR[char]:
                         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:
                 find_name_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)
                         find_name_by_key = False
                         break
-                if find_name_by_key and len(name) > 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 "")
+                if find_name_by_key and name_len > 2:
                     matches = re.findall(r, layout[1])
                     for match in matches:
                         if match in layout[1]: