MMDet——给Dataset类传入logger

首先在train.py中给dataset加入log相关参数

cfg.data.train['log_level'] = cfg.log_level
cfg.data.train['log_file'] = log_file

加入库：

from mmcv.utils import print_log
from mmdet.utils import get_root_logger

有两种log的方法：

• logger.info()：

self.logger.info(f'Distributed training: {distributed}')

• print_log(xxxx, logger=logger)：

print_log(f'All dataset has {len(data["images"])} images', logger=self.logger)