From 15a465a6d12f9e03630c9d3003f51628fae60025 Mon Sep 17 00:00:00 2001
From: FEIJINTI <83849113+FEIJINTI@users.noreply.github.com>
Date: Sat, 6 Aug 2022 12:32:37 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E4=BA=86=E9=A2=84=E6=B5=8B?=
 =?UTF-8?q?=E4=BD=86=E4=B8=8D=E8=BE=93=E5=87=BA?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 main.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/main.py b/main.py
index 75c77e8..d76fc35 100755
--- a/main.py
+++ b/main.py
@@ -58,9 +58,11 @@ def main(only_spec=False, only_color=False):
         if only_spec:
             # 光谱识别
             mask_spec = spec_detector.predict(img_data)
+            mask_rgb = rgb_detector.predict(rgb_data)
             mask_rgb = np.zeros_like(mask_spec, dtype=np.uint8)
         elif only_color:
             # rgb识别
+            mask_spec = spec_detector.predict(img_data)
             mask_rgb = rgb_detector.predict(rgb_data)
             mask_spec = np.zeros_like(mask_rgb, dtype=np.uint8)
         else: