From 606e5fbe7b056735c5945f28a58a93623efc3ddb Mon Sep 17 00:00:00 2001
From: TG <905865530@qq.com>
Date: Thu, 27 Jun 2024 09:38:57 +0800
Subject: [PATCH] =?UTF-8?q?fix:=E4=BF=AE=E5=A4=8D=E4=B8=80=E7=82=B9?=
 =?UTF-8?q?=E5=B0=8F=E7=91=95=E7=96=B5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 20240529RGBtest3/classifer.py | 2 +-
 20240529RGBtest3/main.py      | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/20240529RGBtest3/classifer.py b/20240529RGBtest3/classifer.py
index a2853ff..9e4dd11 100644
--- a/20240529RGBtest3/classifer.py
+++ b/20240529RGBtest3/classifer.py
@@ -382,7 +382,7 @@ class Spec_predict(object):
         '''
         # 对数据进行切片，筛选谱段
         #qt_test进行测试时如果读取的是（30，30，224）需要解开注释进行数据切片，筛选谱段
-        data_x = data_x[ :25, :, setting.selected_bands ]
+        # data_x = data_x[ :25, :, setting.selected_bands ]
         # 将筛选后的数据重塑为二维数组，每行代表一个样本
         data_x = data_x.reshape(-1, setting.n_spec_rows * setting.n_spec_cols * setting.n_spec_bands)
         data_y = self.model.predict(data_x)
diff --git a/20240529RGBtest3/main.py b/20240529RGBtest3/main.py
index 0be2577..28b54f4 100644
--- a/20240529RGBtest3/main.py
+++ b/20240529RGBtest3/main.py
@@ -32,7 +32,7 @@ def main(is_debug=False):
     print('系统初始化中...')
     #模型预热
     #与qt_test测试时需要注释掉预热，模型接收尺寸为（25，30，13），qt_test发送的数据为（30，30，224），需要对数据进行切片（classifer.py第385行）
-    # _ = detector.predict(np.ones((setting.n_spec_rows, setting.n_spec_cols, setting.n_spec_bands), dtype=np.uint16))
+    _ = detector.predict(np.ones((setting.n_spec_rows, setting.n_spec_cols, setting.n_spec_bands), dtype=np.uint16))
     # _ = classifier.predict(np.ones((setting.n_rgb_rows, setting.n_rgb_cols, setting.n_rgb_bands), dtype=np.uint8))
     # _, _, _, _, _ =dp.analyze_tomato(cv2.imread(str(setting.tomato_img_dir)))
     # _, _, _, _, _ = dp.analyze_passion_fruit(cv2.imread(str(setting.passion_fruit_img_dir))