调整了ai模型

This commit is contained in:
FEIJINTI 2022-08-25 14:23:06 +08:00
parent a803b1c905
commit 12870b3880

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/best0823.pt' ai_path = 'weights/best0825.pt'
ai_conf_threshold = 0.5 ai_conf_threshold = 0.5
# mask parameter # mask parameter