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