mmdetection训练detr或deformable detr时mAP为0的问题

 出现图中情况,即使为第一轮也不应全为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的如何训练,仍然有待探索。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值