Update log setting

Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
4 years ago · 0e78755513
1 changed files with 2 additions and 1 deletions
--- 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):