MMDetection加载Detr预训练权重不正确导致检测精度为0

1. 问题描述

本文主要记录一个困扰我很久的Bug。我在做一个实验的时候，需要对Detr原版模型进行finetuning。因此我使用了MMDetection框架（3.3.0版本）进行这个实验。但是在加载预训练权重并训练了一轮以后，出现了非常低的检测结果。

我的配置文件是像这样写的：

_base_ = [
    './detr_r50_8xb2-150e_coco.py'
]
model = dict(
    init_cfg=dict(
        type='Pretrained',
        checkpoint='https://download.openmmlab.com/mmdetection/v3.0/detr/detr_r50_8xb2-150e_coco/detr_r50_8xb2-150e_coco_20221023_153551-436d03e8.pth'
    )
)

# dataset settings
train_dataloader = dict(
    dataset=dict(
        ann_file='annotations/instances_val2017.json',
        data_prefix=dict(img='val2017/')))

我运行训练的命令是：

python tools/train.py configs/detr/detr_r50_8xb2-150e_coco_finetune.py 

2. 原因探究

我又通过命令

python tools/test.py configs