From 7366830c41468947c13e91e796b41f87e9ae390d Mon Sep 17 00:00:00 2001
From: MIAOW <3703781@qq.com>
Date: Thu, 22 Sep 2022 15:12:37 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E9=83=A8=E7=BD=B2=E8=B7=AF?=
 =?UTF-8?q?=E5=BE=84=E5=92=8C=E8=AF=B4=E6=98=8E?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

开发时和机器上部署的路径不同，请注意选择pixel_model_path、blk_model_path、rgb_tobacco_model_path、rgb_background_model_path、ai_path
---
 config.py | 17 ++++++++++++-----
 1 file changed, 12 insertions(+), 5 deletions(-)

diff --git a/config.py b/config.py
index fcdc698..eb11103 100644
--- a/config.py
+++ b/config.py
@@ -20,17 +20,24 @@ class Config:
 
     # 光谱模型参数
     blk_size = 4  # 必须是2的倍数，不然会出错
-    pixel_model_path = r"./weights/pixel_2022-08-02_15-22.model"
-    blk_model_path = r"./weights/rf_4x4_c22_20_sen8_9.model"
+    raise NotImplementedError("开发时和机器上部署的路径不同，请注意选择pixel_model_path、blk_model_path后删除本行")
+    # 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
 
     # rgb模型参数
-    rgb_tobacco_model_path = r"weights/tobacco_dt_2022-08-27_14-43.model"
-    rgb_background_model_path = r"weights/background_dt_2022-08-22_22-15.model"
+    raise NotImplementedError("开发时和机器上部署的路径不同，请注意选择rgb_tobacco_model_path、rgb_background_model_path、ai_path后删除本行")
+    # 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_s = 190  # 饱和度的最高允许值
     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
 
     # mask parameter