查看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等等查看配置。

 

posted @ 2022-03-04 15:04  奋斗的小仔  阅读(426)  评论(0编辑  收藏  举报