From 0e78755513b5bf5d84c5ed3e2606d04066492f69 Mon Sep 17 00:00:00 2001
From: shiyu22 <shiyu.chen@zilliz.com>
Date: Wed, 11 May 2022 17:23:51 +0800
Subject: [PATCH] Update log setting

Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
---
 yolov5.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/yolov5.py b/yolov5.py
index 7f81a0d..41c5ad9 100644
--- a/yolov5.py
+++ b/yolov5.py
@@ -6,8 +6,9 @@ from towhee import register
 from towhee.operator import NNOperator
 
 logging.basicConfig()
-logging.getLogger("yolov5").setLevel(logging.WARNING)
 logging.getLogger().setLevel(logging.WARNING)
+logging.getLogger("yolov5").setLevel(logging.WARNING)
+logging.getLogger("towhee.engine").setLevel(logging.WARNING)
 
 @register(output_schema=['boxes', 'classes', 'scores'])
 class Yolov5(NNOperator):