Caffe-SSD 训练 widerface 一直无法学习的原因?

数据:
转换成VOC格式的widerface

模型:
Caffe-SSD

参数:
ssd_pascal.py默认的,只是修改了一下iteration 为10000 和batch size 为8

结果:
loss从一开始的一两百个iteration迅速下降后,就一直保持在5-8左右徘徊。实测图片结果如下:

1.jpeg


基本上就是一堆无效的预测,概率都是在0.06-0.07左右的非常接近的值。

训练中在annotated_data_layer.cpp的load_batch里输出变换前读取的图片和标注,可以对应上。


请问大家有什么建议吗?


 
已邀请:

要回复问题请先登录注册