diff --git a/yolov5.py b/yolov5.py index 768a912..81daa88 100644 --- a/yolov5.py +++ b/yolov5.py @@ -5,10 +5,9 @@ import logging from towhee import register from towhee.operator import NNOperator -logging.basicConfig() +logging.basicConfig(level=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):