From 55c879d0d4549efdb906755c61401fd1c960f129 Mon Sep 17 00:00:00 2001
From: "li.zhenye" <li.zhenye@qq.com>
Date: Tue, 23 Aug 2022 19:54:37 +0800
Subject: [PATCH] =?UTF-8?q?[fix]=20=E4=BF=AE=E5=A4=8Dai=20path?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

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

diff --git a/main.py b/main.py
index 207b8b2..69e7bed 100755
--- a/main.py
+++ b/main.py
@@ -15,7 +15,8 @@ def main(only_spec=False, only_color=False, if_merge=False, interval_time=None,
          single_spec=False, single_color=False):
     spec_detector = SpecDetector(blk_model_path=Config.blk_model_path, pixel_model_path=Config.pixel_model_path)
     rgb_detector = RgbDetector(tobacco_model_path=Config.rgb_tobacco_model_path,
-                               background_model_path=Config.rgb_background_model_path)
+                               background_model_path=Config.rgb_background_model_path,
+                               ai_path=Config.ai_path)
     _, _ = spec_detector.predict(np.ones((Config.nRows, Config.nCols, Config.nBands), dtype=float)*0.4),\
            rgb_detector.predict(np.ones((Config.nRgbRows, Config.nRgbCols, Config.nRgbBands), dtype=np.uint8)*40)
     total_len = Config.nRows * Config.nCols * Config.nBands * 4  # float型变量, 4个字节