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

Merge remote-tracking branch 'originn/master'

Browse Source
This commit is contained in:
FEIJINTI 2022-08-23 10:53:44 +08:00
commit 9b544a38d4
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
config.py
View File

@ -30,7 +30,7 @@ class Config:
threshold_low, threshold_high = 10, 230 threshold_low, threshold_high = 10, 230
threshold_s = 190 # 饱和度的最高允许值 threshold_s = 190 # 饱和度的最高允许值
rgb_size_threshold = 4 # rgb的尺寸限制 rgb_size_threshold = 4 # rgb的尺寸限制
ai_path = 'weights/best.pt' ai_path = 'weights/best0823.pt'
ai_conf_threshold = 0.5 ai_conf_threshold = 0.5
# mask parameter # mask parameter

2
main.py
View File

@ -122,7 +122,7 @@ def main(only_spec=False, only_color=False, if_merge=False, interval_time=None,
if single_spec: if single_spec:
output_fifos = [mask_fifo_path, ] output_fifos = [mask_fifo_path, ]
elif single_color: elif single_color:
output_fifos = [rgb_fifo_path, ] output_fifos = [rgb_mask_fifo_path, ]
else: else:
output_fifos = [mask_fifo_path, rgb_mask_fifo_path] output_fifos = [mask_fifo_path, rgb_mask_fifo_path]
for fifo, mask in zip(output_fifos, masks): for fifo, mask in zip(output_fifos, masks):