[ext] 增加了横向膨胀/横向padding

config.py

@@ -34,6 +34,7 @@ class Config:
     # mask parameter
     target_size = (1024, 1024)  # (Width, Height) of mask
     valve_merge_size = 2  # 每两个喷阀当中有任意一个出现杂质则认为都是杂质
+    valve_horizontal_padding = 3  # 喷阀横向膨胀的尺寸，应该是奇数,3时表示左右各膨胀1
     max_open_valve_limit = 25  # 最大同时开启喷阀限制,按照电流计算，当前的喷阀可以开启的喷阀 600W的电源 / 12V电源 = 50A, 一个阀门1A
 
     # save part

utils.py

@@ -170,7 +170,7 @@ def valve_merge(img: np.ndarray, merge_size: int = 2) -> np.ndarray:
 
 
 def valve_expend(img: np.ndarray) -> np.ndarray:
-    kernel = np.ones((1, 3), np.uint8)
+    kernel = np.ones((1, Config.valve_horizontal_padding), np.uint8)
     img = cv2.dilate(img, kernel, iterations=1)
     return img