Browse Source
Update log setting
Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
main
1 changed files with
1 additions and
2 deletions
-
yolov5.py
|
|
@ -5,10 +5,9 @@ import logging |
|
|
from towhee import register |
|
|
from towhee import register |
|
|
from towhee.operator import NNOperator |
|
|
from towhee.operator import NNOperator |
|
|
|
|
|
|
|
|
logging.basicConfig() |
|
|
|
|
|
|
|
|
logging.basicConfig(level=logging.WARNING) |
|
|
logging.getLogger().setLevel(logging.WARNING) |
|
|
logging.getLogger().setLevel(logging.WARNING) |
|
|
logging.getLogger("yolov5").setLevel(logging.WARNING) |
|
|
logging.getLogger("yolov5").setLevel(logging.WARNING) |
|
|
# logging.getLogger("towhee.engine").setLevel(logging.WARNING) |
|
|
|
|
|
|
|
|
|
|
|
@register(output_schema=['boxes', 'classes', 'scores']) |
|
|
@register(output_schema=['boxes', 'classes', 'scores']) |
|
|
class Yolov5(NNOperator): |
|
|
class Yolov5(NNOperator): |
|
|
|
