logo
Browse Source

Update logger level

Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
main
shiyu22 3 years ago
parent
commit
2e78e1fc31
  1. 1
      yolov5.py

1
yolov5.py

@ -6,6 +6,7 @@ from towhee import register
from towhee.operator import NNOperator from towhee.operator import NNOperator
logging.getLogger("yolov5").setLevel(logging.WARNING) logging.getLogger("yolov5").setLevel(logging.WARNING)
logging.getLogger().setLevel(logging.WARNING)
@register(output_schema=['boxes', 'classes', 'scores']) @register(output_schema=['boxes', 'classes', 'scores'])
class Yolov5(NNOperator): class Yolov5(NNOperator):

Loading…
Cancel
Save