From 5435723b4c44e8cb50700504371f03fc0bf90c2e Mon Sep 17 00:00:00 2001 From: "li.zhenye" Date: Mon, 22 Aug 2022 19:09:03 +0800 Subject: [PATCH 1/2] =?UTF-8?q?[fix]=20=E4=BF=AE=E5=A4=8D=E4=BA=86?= =?UTF-8?q?=E4=B8=80=E4=B8=AA=E5=8F=AF=E8=83=BD=E5=87=BA=E7=8E=B0bug?= =?UTF-8?q?=E7=9A=84=E5=9C=B0=E6=96=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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): From 10a0a2b23aba90228ce99a658c02e6afa71106aa Mon Sep 17 00:00:00 2001 From: "li.zhenye" Date: Tue, 23 Aug 2022 10:46:47 +0800 Subject: [PATCH 2/2] =?UTF-8?q?[fix]=20=E6=9B=B4=E6=8D=A2=E6=A8=A1?= =?UTF-8?q?=E5=9E=8B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/config.py b/config.py index d93b3e3..718bf17 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