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