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

添加部署路径和说明

Browse Source 
开发时和机器上部署的路径不同，请注意选择pixel_model_path、blk_model_path、rgb_tobacco_model_path、rgb_background_model_path、ai_path
This commit is contained in:
MIAOW 2022-09-22 15:12:37 +08:00 committed by GitHub
parent b2f5cf276c
commit 7366830c41
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
config.py
View File

@ -20,17 +20,24 @@ class Config:
# 光谱模型参数 # 光谱模型参数
blk_size = 4 # 必须是2的倍数，不然会出错 blk_size = 4 # 必须是2的倍数，不然会出错
pixel_model_path = r"./weights/pixel_2022-08-02_15-22.model" raise NotImplementedError("开发时和机器上部署的路径不同，请注意选择pixel_model_path、blk_model_path后删除本行")
blk_model_path = r"./weights/rf_4x4_c22_20_sen8_9.model" # pixel_model_path = r"./weights/pixel_2022-08-02_15-22.model" # 开发时的路径
# pixel_model_path = r"/home/dt/tobacco-color/weights/pixel_2022-08-02_15-22.model" # 机器上部署的路径
# blk_model_path = r"./weights/rf_4x4_c22_20_sen8_9.model" # 开发时的路径
# blk_model_path = r"/home/dt/tobacco-color/weights/rf_4x4_c22_20_sen8_9.model" # 机器上部署的路径
spec_size_threshold = 3 spec_size_threshold = 3
# rgb模型参数 # rgb模型参数
rgb_tobacco_model_path = r"weights/tobacco_dt_2022-08-27_14-43.model" raise NotImplementedError("开发时和机器上部署的路径不同，请注意选择rgb_tobacco_model_path、rgb_background_model_path、ai_path后删除本行")
rgb_background_model_path = r"weights/background_dt_2022-08-22_22-15.model" # rgb_tobacco_model_path = r"weights/tobacco_dt_2022-08-27_14-43.model" # 开发时的路径
# rgb_tobacco_model_path = r"/home/dt/tobacco-color/weights/tobacco_dt_2022-08-27_14-43.model" # 机器上部署的路径
# rgb_background_model_path = r"weights/background_dt_2022-08-22_22-15.model" # 开发时的路径
# rgb_background_model_path = r"/home/dt/tobacco-color/weights/background_dt_2022-08-22_22-15.model" # 机器上部署的路径
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/best0827.pt' # ai_path = 'weights/best0827.pt' # 开发时的路径
# ai_path = '/home/dt/tobacco-color/weights/best0827.pt' # 机器上部署的路径
ai_conf_threshold = 0.6 ai_conf_threshold = 0.6
# mask parameter # mask parameter