search

bubbliiiing / mobilenet-ssd-keras

这是一个mobilenet-ssd-keras的源码，可以用于训练自己的轻量级ssd模型。
MIT License
105 stars 34 forks source link

训练VOC2012，损失最终为3.+，在测试集上漏检很多 #4

Open Rebeccayoyo opened 4 years ago

Rebeccayoyo commented 4 years ago

重新训练的VOC2012数据集，epoch200，初始学习率10-3，取消了早停法。损失最后只能降到3.+，在测试集上表现就是漏检很多。针对模型训练想问以下两个问题。 （1）训练voc数据集，需要加载预训练权重吗？（是加载mobilenet在ImageNet上的预训练权重吗？） （2）作者给出的预训练权重是在VOC2007数据集训练后保存的模型吗？一般应用到其他小数据集需要加载的权重？

bubbliiiing commented 4 years ago

1、都行 2、2007+2012，加载吧 07+12 这个库map66左右 下面的库map73左右 论文版本的Mobilenet-SSD 原版的Mobilenet-ssd不利用38x38的特征层进行回归预测和分类预测，于是我重新弄了一个库，地址如下： https://github.com/bubbliiiing/Mobilenet-SSD-Essay

Rebeccayoyo commented 4 years ago

利用07+12数据集重新训练不利用38*38的特征层的Mobilenet+ssd，训练了120k次，mAP只有58%，这是为什么啊？

bubbliiiing commented 4 years ago

不是预训练权重都有73多么