Skip to content

Commit

Permalink
change logger
Browse files Browse the repository at this point in the history
  • Loading branch information
RunningLeon committed Jan 25, 2022
1 parent 91a9b78 commit bda305d
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions mmdeploy/codebase/mmseg/deploy/segmentation.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,4 @@
# Copyright (c) OpenMMLab. All rights reserved.
import logging
from typing import Any, Dict, Optional, Sequence, Tuple, Union

import mmcv
Expand All @@ -8,7 +7,7 @@
from torch.utils.data import Dataset

from mmdeploy.codebase.base import BaseTask
from mmdeploy.utils import Task, get_input_shape
from mmdeploy.utils import Task, get_input_shape, get_root_logger
from .mmsegmentation import MMSEG_TASK


Expand Down Expand Up @@ -225,7 +224,8 @@ def evaluate_outputs(model_cfg,
to `False`.
"""
if out:
logging.info(f'\nwriting results to {out}')
logger = get_root_logger()
logger.info(f'\nwriting results to {out}')
mmcv.dump(outputs, out)
kwargs = {} if metric_options is None else metric_options
if format_only:
Expand Down

0 comments on commit bda305d

Please sign in to comment.