logo
Browse Source

Update log setting

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

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

Loading…
Cancel
Save