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