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):