[OpenMMLab]使用零碎记录

使用零碎记录

Config

关于config的继承机制，由_base_来完全继承索引的config文件，如果有需要修改的地方，则在_base_下重写即可，例如：

继承的文件中有model如下所示：

model = dict(
    type='ImageClassifier',
    backbone=dict(
        type='ResNet',
        depth=50,
        num_stages=4,
        out_indices=(3, ),
        style='pytorch'),
    neck=dict(type='GlobalAveragePooling'),
    head=dict(
        type='LinearClsHead',
        num_classes=1000,
        in_channels=2048,
        loss=dict(type='CrossEntropyLoss', loss_weight=1.0),
        topk=(1, 5),
    ))

我们需要更改其中的head部分输出类别为30种，则只需把对重新赋值即可，不需要把model或者head部分整体调出重写：

_base_ = ['./configs/_base_/models/resnet50.py']
model = dict(
    head=dict(num_classes=30)
)

此时输出的配置文件为：

model = dict(
    type='ImageClassifier',
    backbone=dict(
        type='ResNet',
        depth=50,
        num_stages=4,
        out_indices=(3, ),
        style='pytorch'),
    neck=dict(type='GlobalAveragePooling'),
    head=dict(
        type='LinearClsHead',
        num_classes=30,
        in_channels=2048,
        loss=dict(type='CrossEntropyLoss', loss_weight=1.0),
        topk=(1, 5)))

发现model[head[numclasses]]已经修改完毕