KeyError: ‘`cfg` or `default_args` must contain the key “type“, but got {\‘num_classes\‘: 2}\nNone‘

最新推荐文章于 2024-10-02 10:53:34 发布

代码飞走咯

最新推荐文章于 2024-10-02 10:53:34 发布

阅读量452

点赞数 4

文章标签： python 深度学习机器学习

本文链接：https://blog.csdn.net/kisushotto/article/details/137968315

版权

解决使用mmdetection库的faster rcnn模型训练时报错

这个报错是由于没有正确重写model cfg所致的：

旧（重写类别数为2）：

model = dict(
    roi_head=dict(
        bbox_head=dict(num_classes=2),
        mask_head=dict(num_classes=2)))

新：

model = dict(
    roi_head=dict(
        type='StandardRoIHead',
        bbox_roi_extractor=dict(
            type='SingleRoIExtractor',
            roi_layer=dict(type='RoIAlign', output_size=7, sampling_ratio=0),
            out_channels=256,
            featmap_strides=[4, 8, 16, 32]),
        bbox_head=dict(
            type='Shared2FCBBoxHead',
            in_channels=256,
            fc_out_channels=1024,
            roi_feat_size=7,
            num_classes=2,
            bbox_coder=dict(
                type='DeltaXYWHBBoxCoder',
                target_means=[0., 0., 0., 0.],
                target_stds=[0.1, 0.1, 0.2, 0.2]),
            reg_class_agnostic=False,
            loss_cls=dict(
                type='CrossEntropyLoss', use_sigmoid=False, loss_weight=1.0),
            loss_bbox=dict(type='L1Loss', loss_weight=1.0)))
)