diff --git a/config.py b/config.py
index eb8e4e1..10e30da 100644
--- a/config.py
+++ b/config.py
@@ -30,7 +30,7 @@ class Config:
     threshold_low, threshold_high = 10, 230
     threshold_s = 190  # 饱和度的最高允许值
     rgb_size_threshold = 4  # rgb的尺寸限制
-    ai_path = 'weights/best.pt'
+    ai_path = 'weights/best0823.pt'
     ai_conf_threshold = 0.5
 
     # mask parameter
diff --git a/main.py b/main.py
index 741d4cf..207b8b2 100755
--- a/main.py
+++ b/main.py
@@ -122,7 +122,7 @@ def main(only_spec=False, only_color=False, if_merge=False, interval_time=None,
         if single_spec:
             output_fifos = [mask_fifo_path, ]
         elif single_color:
-            output_fifos = [rgb_fifo_path, ]
+            output_fifos = [rgb_mask_fifo_path, ]
         else:
             output_fifos = [mask_fifo_path, rgb_mask_fifo_path]
         for fifo, mask in zip(output_fifos, masks):