出现图中情况,即使为第一轮也不应全为0
数据集设置的也没有问题,刚刚还跑过SSD和YOLOX
在csdn上查询,有人说是预训练权重的原因。对resnet预训练权重修改就可以解决问题。
MMdetection3.0 训练DETR问题分析_mmdetection 训练detr-优快云博客
我按照其方法修改,没有效果。
也有人说是num_classes设置不对,我正确设置后也未能解决。
经过排查确实是预训练权重有问题。
我的解决方法:
在mmdetection/configs/detr at main · open-mmlab/mmdetection · GitHub,上下载mmdetection官方提供的detr-r50预训练权重,修改配置文件加载它。
问题成功解决。
需要注意的是,mmdet官方只提供了detr-r50.的预训练权重,至于r18和r101的如何训练,仍然有待探索。