查看mmdetection中模型的配置信息
方法一
可以直接打开mmdetection中的目录查看,/configs目录下都是对应的模型的配置
示例:
可以找到_base_目录下的这四个文件文件查看配置。
方法二
读取配置文件查看
在命令行中输入(当前目录是mmdetection)
>>> from mmcv import Config >>> cfg = Config.fromfile('configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py')
配置信息便以字典的形式保存在了cfg变量中。我们可以输入对应的命令来查看配置信息。
1 >>> cfg.dataset_type 2 'CocoDataset' 3 >>> cfg.data_root 4 'data/coco/' 5 >>> cfg.workflow 6 [('train', 1)] 7 >>> cfg.lr_config 8 {'policy': 'step', 'warmup': 'linear', 'warmup_iters': 500, 'warmup_ratio': 0.001, 'step': [8, 11]} 9 >>> cfg.runner 10 {'type': 'EpochBasedRunner', 'max_epochs': 12} 11 >>> cfg.optimizer 12 {'type': 'SGD', 'lr': 0.02, 'momentum': 0.9, 'weight_decay': 0.0001}
还可以输入cfg.data,cfg.model等等查看配置。