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

This commit is contained in:
li.zhenye 2022-08-09 11:46:16 +08:00
parent 3cdafcd6fd
commit 15a126a3ea
2 changed files with 2 additions and 1 deletions

View File

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

View File

@ -170,7 +170,7 @@ def valve_merge(img: np.ndarray, merge_size: int = 2) -> np.ndarray:
def valve_expend(img: np.ndarray) -> 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) img = cv2.dilate(img, kernel, iterations=1)
return img return img