Ruoqi Li
Ruoqi Li
> 博主,你好,我在用VOC训练的时候,将num_class设置成20,然后去训练得到的map是0.63左右,然后您提供的78.7.pth测试的时候也需要将num_classes设置为80才能测试成功,这是什么原因呢 请问您找到原因了吗?我也遇到了同样的问题
> > > 博主,你好,我在用VOC训练的时候,将num_class设置成20,然后去训练得到的map是0.63左右,然后您提供的78.7.pth测试的时候也需要将num_classes设置为80才能测试成功,这是什么原因呢 > > > > > > 请问您找到原因了吗?我也遇到了同样的问题 > > 我感觉代码是没有问题的,我基本上复现了mAP精度。作者提供的pth文件可能是在训练时没有改class_num,所有导致测试VOC需要将其设为80,但是自己从头训练的话只需要设为20或者21(with background)就行了。 作者提到VOC训练30epo,虽然看loss下降趋势感觉后续训练没有提升模型,但是最好还是按照30epo来训练。因为loss里面包含了centerness loss,loss下降不明显主要是这一部分的原因,可以在训练时单独把classification+regression loss 拎出来显示。 非常感谢回复!可以请问一下您是怎么提高mAP的吗?所有设置都和作者提供的一样吗?我用两张卡训的,所以batchsize小了,也适当降低了lr,但是训练了30甚至50个epoch后,mAP只有40+。
> > > > > 博主,你好,我在用VOC训练的时候,将num_class设置成20,然后去训练得到的map是0.63左右,然后您提供的78.7.pth测试的时候也需要将num_classes设置为80才能测试成功,这是什么原因呢 > > > > > > > > > > > > 请问您找到原因了吗?我也遇到了同样的问题 > > > > > > > > > 我感觉代码是没有问题的,我基本上复现了mAP精度。作者提供的pth文件可能是在训练时没有改class_num,所有导致测试VOC需要将其设为80,但是自己从头训练的话只需要设为20或者21(with background)就行了。...