qiexing / face-landmark-localization

cnn network predict face landmarks (68 points) and head pose (3d pose, yaw,roll,pitch).
410 stars 184 forks source link

训练loss停止下降问题 #24

Open sophieyutingye opened 7 years ago

sophieyutingye commented 7 years ago

作者您好,您的工作很棒,我想跟着你用你train文件夹里面的网络结构自己重新训练一下,但是出现了了loss下降到0.7左右时候就没法下降,不论人脸的姿势是怎么样的,训练检测的每张图片的脸的68个点几乎都是端正分布在图中构成一个正面的脸。

请问您训练的过程中有遇到过这个问题吗?针对这个问题您有没有什么意见或建议以便我能寻找这个问题的根源呢,谢谢!期待您的回复

sophieyutingye commented 7 years ago

![Uploading image.png…]() 345

qiexing commented 7 years ago

@sophieyutingye 如果你没有用imagenet pre-trained model的话,可能是网络的权重初始化造成的。推荐网络用xavier初始化。比如:
weight_filler { type: "xavier" variance_norm: AVERAGE

}

这个语法不一定适用现在的caffe,可以查询一下最新的caffe文档。

sophieyutingye commented 7 years ago

非常感谢你的回答 我改了下感觉有很大的进步!!祝生活愉快,学业有成!

AlphaQi commented 7 years ago

@sophieyutingye 你好,请问你是复现了这个对吧?能否告诉下你用到了哪些数据集扩增方法,非常感谢