mmdetection配置文件-参数含义

mmdetection配置文件-参数含义

可以参考下面mmdetection官网文档：如下网址

https://github.com/open-mmlab/mmdetection/blob/master/docs/zh_cn/tutorials/config.md

因为部分人可能访问这个github上的网址速度较慢，所以引用这个文档来介绍配置文件各参数。

–如有侵权，立即删除

为了帮助用户对 MMDetection 检测系统中的完整配置和模块有一个基本的了解，官方对使用 ResNet50 和 FPN 的 Mask R-CNN 的配置文件进行简要注释说明。更详细的用法和各个模块对应的替代方案，请参考 API 文档。

Mask R-CNN配置文件示例

model = dict(type='MaskRCNN',  # 检测器(detector)名称backbone=dict(  # 主干网络的配置文件type='ResNet',  # 主干网络的类别，可用选项请参考 https://github.com/open-mmlab/mmdetection/blob/master/mmdet/models/backbones/resnet.py#L308depth=50,  # 主干网络的深度，对于 ResNet 和 ResNext 通常设置为 50 或 101。num_stages=4,  # 主干网络状态(stages)的数目，这些状态产生的特征图作为后续的 head 的输入。out_indices=(0, 1, 2, 3),  # 每个状态产生的特征图输出的索引。frozen_stages=1,  # 第一个状态的权重被冻结norm_cfg=dict(  # 归一化层(norm layer)的配置项。type='BN',  # 归一化层的类别，通常是 BN 或 GN。requires_grad=True),  # 是否训练归一化里的 gamma 和 beta。norm_eval=True,  # 是否冻结 BN 里的统计项。style='pytorch',  # 主干网络的风格，'pytorch' 意思是步长为2的层为 3x3 卷积， 'caffe' 意思是步长为2的层为 1x1 卷积。init_cfg=dict(type='Pretrained', checkpoint='torchvision://resnet50')),  # 加载通过 ImageNet 预训练的模型neck=dict(type='FPN',  # 检测器的 neck 是 FPN，我们同样支持 'NASFPN', 'PAFPN' 等，更多细节可以参考 https://github.com/open-mmlab/mmdetection/blob/master/mmdet/models/necks/fpn.py#L10。in_channels=[256, 512, 1024, 2048],  # 输入通道数，这与主干网络的输出通道一致out_channels=256,  # 金字塔特征图每一层的输出通道num_outs=5),  # 输出的范围(scales)rpn_head=dict(type='RPNHead',  # RPN_head 的类型是 'RPNHead', 我们也支持 'GARPNHead' 等，更多细节可以参考 https://github.com/open-mmlab/mmdetection/blob/master/mmdet/models/dense_heads/rpn_head.py#L12。in_channels=256,  # 每个输入特征图的输入通道，这与 neck 的输出通道一致。feat_channels=256,  # head 卷积层的特征通道。anchor_generator=dict(  # 锚点(Anchor)生成器的配置。type='AnchorGenerator',  # 大多是方法使用 AnchorGenerator 作为锚点生成器, SSD 检测器使用 `SSDAnchorGenerator`。更多细节请参考 https://github.com/open-mmlab/mmdetection/blob/master/mmdet/core/anchor/anchor_generator.py#L10。scales=[8],  # 锚点的基本比例，特征图某一位置的锚点面积为 scale * base_sizesratios=[0.5, 1.0, 2.0],  # 高度和宽度之间的比率。strides=[4, 8, 16, 32, 64]),  # 锚生成器的步幅。这与 FPN 特征步幅一致。 如果未设置 base_sizes，则当前步幅值将被视为 base_sizes。bbox_coder=dict(  # 在训练和测试期间对框进行编码和解码。type='DeltaXYWHBBoxCoder',  # 框编码器的类别，'DeltaXYWHBBoxCoder' 是最常用的，更多细节请参考 https://github.com/open-mmlab/mmdetection/blob/master/mmdet/core/bbox/coder/delta_xywh_bbox_coder.py#L9。target_means=[0.0, 0.0, 0.0, 0.0],  # 用于编码和解码框的目标均值target_stds=[1.0, 1.0, 1.0, 1.0]),  # 用于编码和解码框的标准差loss_cls=dict(  # 分类分支的损失函数配置type='CrossEntropyLoss',  # 分类分支的损失类型，我们也支持 FocalLoss 等。use_sigmoid=True,  # RPN通常进行二分类，所以通常使用sigmoid函数。los_weight=1.0),  # 分类分支的损失权重。loss_bbox=dict(  # 回归分支的损失函数配置。type='L1Loss',  # 损失类型，我们还支持许多 IoU Losses 和 Smooth L1-loss 等，更多细节请参考 https://github.com/open-mmlab/mmdetection/blob/master/mmdet/models/losses/smooth_l1_loss.py#L56。loss_weight=1.0)),  # 回归分支的损失权重。roi_head=dict(  # RoIHead 封装了两步(two-stage)/级联(cascade)检测器的第二步。type='StandardRoIHead',  # RoI head 的类型，更多细节请参考 https://github.com/open-mmlab/mmdetection/blob/master/mmdet/models/roi_heads/standard_roi_head.py#L10。bbox_roi_extractor=dict(  # 用于 bbox 回归的 RoI 特征提取器。type='SingleRoIExtractor',  # RoI 特征提取器的类型，大多数方法使用  SingleRoIExtractor，更多细节请参考 https://github.com/open-mmlab/mmdetection/blob/master/mmdet/models/roi_heads/roi_extractors/single_level.py#L10。roi_layer=dict(  # RoI 层的配置type='RoIAlign',  # RoI 层的类别, 也支持 DeformRoIPoolingPack 和 ModulatedDeformRoIPoolingPack，更多细节请参考 https://github.com/open-mmlab/mmdetection/blob/master/mmdet/ops/roi_align/roi_align.py#L79。output_size=7,  # 特征图的输出大小。sampling_ratio=0),  # 提取 RoI 特征时的采样率。0 表示自适应比率。out_channels=256,  # 提取特征的输出通道。featmap_strides=[4, 8, 16, 32]),  # 多尺度特征图的步幅，应该与主干的架构保持一致。bbox_head=dict(  # RoIHead 中 box head 的配置.type='Shared2FCBBoxHead',  # bbox head 的类别，更多细节请参考 https://github.com/open-mmlab/mmdetection/blob/master/mmdet/models/roi_heads/bbox_heads/convfc_bbox_head.py#L177。in_channels=256,  # bbox head 的输入通道。 这与 roi_extractor 中的 out_channels 一致。fc_out_channels=1024,  # FC 层的输出特征通道。roi_feat_size=7,  # 候选区域(Region of Interest)特征的大小。num_classes=80,  # 分类的类别数量。bbox_coder=dict(  # 第二阶段使用的框编码器。type='DeltaXYWHBBoxCoder',  # 框编码器的类别，大多数情况使用 'DeltaXYWHBBoxCoder'。target_means=[0.0, 0.0, 0.0, 0.0],  # 用于编码和解码框的均值target_stds=[0.1, 0.1, 0.2, 0.2]),  # 编码和解码的标准差。因为框更准确，所以值更小，常规设置时 [0.1, 0.1, 0.2, 0.2]。reg_class_agnostic=False,  # 回归是否与类别无关。loss_cls=dict(  # 分类分支的损失函数配置type='CrossEntropyLoss',  # 分类分支的损失类型，我们也支持 FocalLoss 等。use_sigmoid=False,  # 是否使用 sigmoid。loss_weight=1.0),  # 分类分支的损失权重。loss_bbox=dict(  # 回归分支的损失函数配置。type='L1Loss',  # 损失类型，我们还支持许多 IoU Losses 和 Smooth L1-loss 等。loss_weight=1.0)),  # 回归分支的损失权重。mask_roi_extractor=dict(  # 用于 mask 生成的 RoI 特征提取器。type='SingleRoIExtractor',  # RoI 特征提取器的类型，大多数方法使用 SingleRoIExtractor。roi_layer=dict(  # 提取实例分割特征的 RoI 层配置type='RoIAlign',  # RoI 层的类型，也支持 DeformRoIPoolingPack 和 ModulatedDeformRoIPoolingPack。output_size=14,  # 特征图的输出大小。sampling_ratio=0),  # 提取 RoI 特征时的采样率。out_channels=256,  # 提取特征的输出通道。featmap_strides=[4, 8, 16, 32]),  # 多尺度特征图的步幅。mask_head=dict(  # mask 预测 head 模型type='FCNMaskHead',  # mask head 的类型，更多细节请参考 https://github.com/open-mmlab/mmdetection/blob/master/mmdet/models/roi_heads/mask_heads/fcn_mask_head.py#L21。num_convs=4,  # mask head 中的卷积层数in_channels=256,  # 输入通道，应与 mask roi extractor 的输出通道一致。conv_out_channels=256,  # 卷积层的输出通道。num_classes=80,  # 要分割的类别数。loss_mask=dict(  # mask 分支的损失函数配置。type='CrossEntropyLoss',  # 用于分割的损失类型。use_mask=True,  # 是否只在正确的类中训练 mask。loss_weight=1.0))))  # mask 分支的损失权重.train_cfg = dict(  # rpn 和 rcnn 训练超参数的配置rpn=dict(  # rpn 的训练配置assigner=dict(  # 分配器(assigner)的配置type='MaxIoUAssigner',  # 分配器的类型，MaxIoUAssigner 用于许多常见的检测器，更多细节请参考 https://github.com/open-mmlab/mmdetection/blob/master/mmdet/core/bbox/assigners/max_iou_assigner.py#L10。pos_iou_thr=0.7,  # IoU >= 0.7(阈值) 被视为正样本。neg_iou_thr=0.3,  # IoU < 0.3(阈值) 被视为负样本。min_pos_iou=0.3,  # 将框作为正样本的最小 IoU 阈值。match_low_quality=True,  # 是否匹配低质量的框(更多细节见 API 文档).ignore_iof_thr=-1),  # 忽略 bbox 的 IoF 阈值。sampler=dict(  # 正/负采样器(sampler)的配置type='RandomSampler',  # 采样器类型，还支持 PseudoSampler 和其他采样器，更多细节请参考 https://github.com/open-mmlab/mmdetection/blob/master/mmdet/core/bbox/samplers/random_sampler.py#L8。num=256,  # 样本数量。pos_fraction=0.5,  # 正样本占总样本的比例。neg_pos_ub=-1,  # 基于正样本数量的负样本上限。add_gt_as_proposals=False),  # 采样后是否添加 GT 作为 proposal。allowed_border=-1,  # 填充有效锚点后允许的边框。pos_weight=-1,  # 训练期间正样本的权重。debug=False),  # 是否设置调试(debug)模式rpn_proposal=dict(  # 在训练期间生成 proposals 的配置nms_across_levels=False,  # 是否对跨层的 box 做 NMS。仅适用于 `GARPNHead` ，naive rpn 不支持 nms cross levels。nms_pre=2000,  # NMS 前的 box 数nms_post=1000,  # NMS 要保留的 box 的数量，只在 GARPNHHead 中起作用。max_per_img=1000,  # NMS 后要保留的 box 数量。nms=dict( # NMS 的配置type='nms',  # NMS 的类别iou_threshold=0.7 # NMS 的阈值),min_bbox_size=0),  # 允许的最小 box 尺寸rcnn=dict(  # roi head 的配置。assigner=dict(  # 第二阶段分配器的配置，这与 rpn 中的不同type='MaxIoUAssigner',  # 分配器的类型，MaxIoUAssigner 目前用于所有 roi_heads。更多细节请参考 https://github.com/open-mmlab/mmdetection/blob/master/mmdet/core/bbox/assigners/max_iou_assigner.py#L10。pos_iou_thr=0.5,  # IoU >= 0.5(阈值)被认为是正样本。neg_iou_thr=0.5,  # IoU < 0.5(阈值)被认为是负样本。min_pos_iou=0.5,  # 将 box 作为正样本的最小 IoU 阈值match_low_quality=False,  # 是否匹配低质量下的 box(有关更多详细信息，请参阅 API 文档)。ignore_iof_thr=-1),  # 忽略 bbox 的 IoF 阈值sampler=dict(type='RandomSampler',  #采样器的类型，还支持 PseudoSampler 和其他采样器，更多细节请参考 https://github.com/open-mmlab/mmdetection/blob/master/mmdet/core/bbox/samplers/random_sampler.py#L8。num=512,  # 样本数量pos_fraction=0.25,  # 正样本占总样本的比例。.neg_pos_ub=-1,  # 基于正样本数量的负样本上限。.add_gt_as_proposals=True),  # 采样后是否添加 GT 作为 proposal。mask_size=28,  # mask 的大小pos_weight=-1,  # 训练期间正样本的权重。debug=False))  # 是否设置调试模式。test_cfg = dict(  # 用于测试 rpn 和 rcnn 超参数的配置rpn=dict(  # 测试阶段生成 proposals 的配置nms_across_levels=False,  # 是否对跨层的 box 做 NMS。仅适用于`GARPNHead`，naive rpn 不支持做 NMS cross levels。nms_pre=1000,  # NMS 前的 box 数nms_post=1000,  # NMS 要保留的 box 的数量，只在`GARPNHHead`中起作用。max_per_img=1000,  # NMS 后要保留的 box 数量nms=dict( # NMS 的配置type='nms',  # NMS 的类型iou_threshold=0.7 # NMS 阈值),min_bbox_size=0),  # box 允许的最小尺寸rcnn=dict(  # roi heads 的配置score_thr=0.05,  # bbox 的分数阈值nms=dict(  # 第二步的 NMS 配置type='nms',  # NMS 的类型iou_thr=0.5),  # NMS 的阈值max_per_img=100,  # 每张图像的最大检测次数mask_thr_binary=0.5))  # mask 预处的阈值
dataset_type = 'CocoDataset'  # 数据集类型，这将被用来定义数据集。
data_root = 'data/coco/'  # 数据的根路径。
img_norm_cfg = dict(  # 图像归一化配置，用来归一化输入的图像。mean=[123.675, 116.28, 103.53],  # 预训练里用于预训练主干网络模型的平均值。std=[58.395, 57.12, 57.375],  # 预训练里用于预训练主干网络模型的标准差。to_rgb=True
)  #  预训练里用于预训练主干网络的图像的通道顺序。
train_pipeline = [  # 训练流程dict(type='LoadImageFromFile'),  # 第 1 个流程，从文件路径里加载图像。dict(type='LoadAnnotations',  # 第 2 个流程，对于当前图像，加载它的注释信息。with_bbox=True,  # 是否使用标注框(bounding box)， 目标检测需要设置为 True。with_mask=True,  # 是否使用 instance mask，实例分割需要设置为 True。poly2mask=False),  # 是否将 polygon mask 转化为 instance mask, 设置为 False 以加速和节省内存。dict(type='Resize',  # 变化图像和其注释大小的数据增广的流程。img_scale=(1333, 800),  # 图像的最大规模。keep_ratio=True),  # 是否保持图像的长宽比。dict(type='RandomFlip',  #  翻转图像和其注释大小的数据增广的流程。flip_ratio=0.5),  # 翻转图像的概率。dict(type='Normalize',  # 归一化当前图像的数据增广的流程。mean=[123.675, 116.28, 103.53],  # 这些键与 img_norm_cfg 一致，因为 img_norm_cfg 被std=[58.395, 57.12, 57.375],     # 用作参数。to_rgb=True),dict(type='Pad',  # 填充当前图像到指定大小的数据增广的流程。size_divisor=32),  # 填充图像可以被当前值整除。dict(type='DefaultFormatBundle'),  # 流程里收集数据的默认格式捆。dict(type='Collect',  # 决定数据中哪些键应该传递给检测器的流程keys=['img', 'gt_bboxes', 'gt_labels', 'gt_masks'])
]
test_pipeline = [dict(type='LoadImageFromFile'),  # 第 1 个流程，从文件路径里加载图像。dict(type='MultiScaleFlipAug',  # 封装测试时数据增广(test time augmentations)。img_scale=(1333, 800),  # 决定测试时可改变图像的最大规模。用于改变图像大小的流程。flip=False,  # 测试时是否翻转图像。transforms=[dict(type='Resize',  # 使用改变图像大小的数据增广。keep_ratio=True),  # 是否保持宽和高的比例，这里的图像比例设置将覆盖上面的图像规模大小的设置。dict(type='RandomFlip'),  # 考虑到 RandomFlip 已经被添加到流程里，当 flip=False 时它将不被使用。dict(type='Normalize',  #  归一化配置项，值来自 img_norm_cfg。mean=[123.675, 116.28, 103.53],std=[58.395, 57.12, 57.375],to_rgb=True),dict(type='Pad',  # 将配置传递给可被 32 整除的图像。size_divisor=32),dict(type='ImageToTensor',  # 将图像转为张量keys=['img']),dict(type='Collect',  # 收集测试时必须的键的收集流程。keys=['img'])])
]
data = dict(samples_per_gpu=2,  # 单个 GPU 的 Batch sizeworkers_per_gpu=2,  # 单个 GPU 分配的数据加载线程数train=dict(  # 训练数据集配置type='CocoDataset',  # 数据集的类别, 更多细节请参考 https://github.com/open-mmlab/mmdetection/blob/master/mmdet/datasets/coco.py#L19。ann_file='data/coco/annotations/instances_train2017.json',  # 注释文件路径img_prefix='data/coco/train2017/',  # 图片路径前缀pipeline=[  # 流程, 这是由之前创建的 train_pipeline 传递的。dict(type='LoadImageFromFile'),dict(type='LoadAnnotations',with_bbox=True,with_mask=True,poly2mask=False),dict(type='Resize', img_scale=(1333, 800), keep_ratio=True),dict(type='RandomFlip', flip_ratio=0.5),dict(type='Normalize',mean=[123.675, 116.28, 103.53],std=[58.395, 57.12, 57.375],to_rgb=True),dict(type='Pad', size_divisor=32),dict(type='DefaultFormatBundle'),dict(type='Collect',keys=['img', 'gt_bboxes', 'gt_labels', 'gt_masks'])]),val=dict(  # 验证数据集的配置type='CocoDataset',ann_file='data/coco/annotations/instances_val2017.json',img_prefix='data/coco/val2017/',pipeline=[  # 由之前创建的 test_pipeline 传递的流程。dict(type='LoadImageFromFile'),dict(type='MultiScaleFlipAug',img_scale=(1333, 800),flip=False,transforms=[dict(type='Resize', keep_ratio=True),dict(type='RandomFlip'),dict(type='Normalize',mean=[123.675, 116.28, 103.53],std=[58.395, 57.12, 57.375],to_rgb=True),dict(type='Pad', size_divisor=32),dict(type='ImageToTensor', keys=['img']),dict(type='Collect', keys=['img'])])]),test=dict(  # 测试数据集配置，修改测试开发/测试(test-dev/test)提交的 ann_filetype='CocoDataset',ann_file='data/coco/annotations/instances_val2017.json',img_prefix='data/coco/val2017/',pipeline=[  # 由之前创建的 test_pipeline 传递的流程。dict(type='LoadImageFromFile'),dict(type='MultiScaleFlipAug',img_scale=(1333, 800),flip=False,transforms=[dict(type='Resize', keep_ratio=True),dict(type='RandomFlip'),dict(type='Normalize',mean=[123.675, 116.28, 103.53],std=[58.395, 57.12, 57.375],to_rgb=True),dict(type='Pad', size_divisor=32),dict(type='ImageToTensor', keys=['img']),dict(type='Collect', keys=['img'])])],samples_per_gpu=2  # 单个 GPU 测试时的 Batch size))
evaluation = dict(  # evaluation hook 的配置，更多细节请参考 https://github.com/open-mmlab/mmdetection/blob/master/mmdet/core/evaluation/eval_hooks.py#L7。interval=1,  # 验证的间隔。metric=['bbox', 'segm'])  # 验证期间使用的指标。
optimizer = dict(  # 用于构建优化器的配置文件。支持 PyTorch 中的所有优化器，同时它们的参数与 PyTorch 里的优化器参数一致。type='SGD',  # 优化器种类，更多细节可参考 https://github.com/open-mmlab/mmdetection/blob/master/mmdet/core/optimizer/default_constructor.py#L13。lr=0.02,  # 优化器的学习率，参数的使用细节请参照对应的 PyTorch 文档。momentum=0.9,  # 动量(Momentum)weight_decay=0.0001)  # SGD 的衰减权重(weight decay)。
optimizer_config = dict(  # optimizer hook 的配置文件，执行细节请参考 https://github.com/open-mmlab/mmcv/blob/master/mmcv/runner/hooks/optimizer.py#L8。grad_clip=None)  # 大多数方法不使用梯度限制(grad_clip)。
lr_config = dict(  # 学习率调整配置，用于注册 LrUpdater hook。policy='step',  # 调度流程(scheduler)的策略，也支持 CosineAnnealing, Cyclic, 等。请从 https://github.com/open-mmlab/mmcv/blob/master/mmcv/runner/hooks/lr_updater.py#L9 参考 LrUpdater 的细节。warmup='linear',  # 预热(warmup)策略，也支持 `exp` 和 `constant`。warmup_iters=500,  # 预热的迭代次数warmup_ratio=0.001,  # 用于热身的起始学习率的比率step=[8, 11])  # 衰减学习率的起止回合数
runner = dict(type='EpochBasedRunner',  # 将使用的 runner 的类别 (例如 IterBasedRunner 或 EpochBasedRunner)。max_epochs=12) # runner 总回合数， 对于 IterBasedRunner 使用 `max_iters`
checkpoint_config = dict(  # Checkpoint hook 的配置文件。执行时请参考 https://github.com/open-mmlab/mmcv/blob/master/mmcv/runner/hooks/checkpoint.py。interval=1)  # 保存的间隔是 1。
log_config = dict(  # register logger hook 的配置文件。interval=50,  # 打印日志的间隔hooks=[# dict(type='TensorboardLoggerHook')  # 同样支持 Tensorboard 日志dict(type='TextLoggerHook')])  # 用于记录训练过程的记录器(logger)。
dist_params = dict(backend='nccl')  # 用于设置分布式训练的参数，端口也同样可被设置。
log_level = 'INFO'  # 日志的级别。
load_from = None  # 从一个给定路径里加载模型作为预训练模型，它并不会消耗训练时间。
resume_from = None  # 从给定路径里恢复检查点(checkpoints)，训练模式将从检查点保存的轮次开始恢复训练。
workflow = [('train', 1)]  # runner 的工作流程，[('train', 1)] 表示只有一个工作流且工作流仅执行一次。根据 total_epochs 工作流训练 12个回合。
work_dir = 'work_dir'  # 用于保存当前实验的模型检查点和日志的目录。

本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！