关于MMDetection训练YOLOv3时出现报错

Question

TypeError: class YOLOV3 in mmdet/models/detectors/yolo.py: init() got an unexpected keyword argument 'roi_head'
用mmdetection训练YOLOv3模型时出现以上报错，在网上找了很久也没有找到解决办法，有没有懂哥知道如何解决吗？万分感谢！！

以下是我的配置文件：

# 新配置继承了基本配置，并做了必要的修改
_base_ = '../yolo/yolov3_d53_8xb8-320-273e_coco.py'

# 我们还需要更改 head 中的 num_classes 以匹配数据集中的类别数
model = dict(
    roi_head=dict(
        bbox_head=dict(num_classes=10), mask_head=dict(num_classes=10)))

# 修改数据集相关配置
data_root = '/home/mmdetection/data/VOCdevkit/VOC2007_COCO/'
metainfo = {
    'classes': ('plane', 'boat', 'camping_car', 'car', 'pick-up', 'tractor', 'truck', 'van',
              'vehicle', 'others'),
    'palette': [
        (106, 0, 228), (119, 11, 32), (165, 42, 42), (0, 0, 192),
                    (197, 226, 255), (0, 60, 100), (0, 0, 142), (255, 77, 255),
                    (153, 69, 1), (120, 166, 157)
    ]
}
train_dataloader = dict(
    batch_size=1,
    dataset=dict(
        data_root=data_root,
        metainfo=metainfo,
        ann_file='annotations/instances_train2017.json',
        data_prefix=dict(img='train2017/')))
val_dataloader = dict(
    dataset=dict(
        data_root=data_root,
        metainfo=metainfo,
        ann_file='annotations/instances_val2017.json',
        data_prefix=dict(img='val2017/')))
test_dataloader = val_dataloader

# 修改评价指标相关配置
val_evaluator = dict(ann_file=data_root + 'annotations/instances_val2017.json')
test_evaluator = val_evaluator

# 使用预训练的 Mask R-CNN 模型权重来做初始化，可以提高模型性能
load_from = 'checkpoints/yolov3_d53_320_273e_coco-421362b6.pth'

Answer 1

报错的意思是YOLOV3 这个类的输入参数里面没有这个roi_head参数，我没用过MMD，需要你自己看下这个类的__init__里面的参数具体是哪些