NanjingForestryUniversity/supermachine-tobacco
3
0
Fork 0
mirror of https://github.com/NanjingForestryUniversity/supermachine-tobacco.git synced 2025-11-08 14:23:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

[ext] 补充修改，关闭喷阀限制功能

Browse Source
This commit is contained in:
li.zhenye 2022-08-11 17:37:52 +08:00
parent 84199c1d75
commit 5754c74167
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
main.py
View File

@ -92,8 +92,8 @@ def main(only_spec=False, only_color=False, if_merge=False, interval_time=None,
mask_rgb = rgb_detector.predict(rgb_data).astype(np.uint8) mask_rgb = rgb_detector.predict(rgb_data).astype(np.uint8)
# 进行多个喷阀的合并 # 进行多个喷阀的合并
masks = [utils.valve_expend(mask) for mask in [mask_spec, mask_rgb]] masks = [utils.valve_expend(mask) for mask in [mask_spec, mask_rgb]]
# 进行喷阀同时开启限制 # 进行喷阀同时开启限制,在8月11日后收到倪超老师的电话，关闭
masks = [utils.valve_limit(mask, Config.max_open_valve_limit) for mask in masks] # masks = [utils.valve_limit(mask, Config.max_open_valve_limit) for mask in masks]
# control the size of the output masks, 在resize前，图像的宽度是和喷阀对应的 # control the size of the output masks, 在resize前，图像的宽度是和喷阀对应的
masks = [cv2.resize(mask.astype(np.uint8), Config.target_size) for mask in masks] masks = [cv2.resize(mask.astype(np.uint8), Config.target_size) for mask in masks]
# merge the masks if needed # merge the masks if needed