zhenghao977 / FCOS-PyTorch-37.2AP

A pure torch implement of FCOS 37.2AP
162 stars 39 forks source link

训练自己数据集,loss不下降但是正确率很高 #21

Open Fuhan1994 opened 2 years ago

Fuhan1994 commented 2 years ago

你好,想请教一下,我训练自己的数据集,totalloss一直在0.6-0.76之间波动,但是验证map最终达到0.99,这和在mmdetection上训练验证结果不同,想请教loss这样合理么?代码中有加入正则化防止过拟合么?非常感谢!

zhenghao977 commented 2 years ago

@Fuhan1994 loss计算与官方release的是一致的 可以检查下在训练数据上的结果。total loss最后收敛在这个位置记忆中是正常的,因为centerness loss大概收敛到0.5。