前段时间使用caffe版本的SSD训练人脸检测,效果还不错,在FDDB上测试了下结果最终只有89%的准确率,但是误判率很低,结果如下:
为了分析训练的SSD在FDDB上对于哪些类型的人脸检测结果较差,结果如下:蓝色的椭圆为FDDB的原始标注,红色的矩形为现有的SSD检测结果。
可以看到:
很模糊的脸没有检测到
还有些基本上算不上脸
极少数是人脸的确实没有检测到
发现FDDB中不少人脸基本可以不算是人脸,因此对FDDB的检测结果的意义有点怀疑了。
最终还是自己建立了专门的评价数据集(共1000多张样本):
少数较为正常的人脸
遮挡较多
不同光照条件下的人脸
角度多变的人脸
检测结果如下(蓝线为检测的人脸框),效果还是不错的:
SSD做检测训练的时候注意:
自己没有必要做简单的数据增强(例如翻转啥的),SSD训练过程中会做数据集增强
训练的数据集,不要用很模糊的人脸,或者很小的人脸,这样会导致训练发散los=Nan如下图(可能),因为SSD训练会根据IOU选取一定比率的样本作为positive样本。
欢迎光临 纳金网 (http://old.narkii.com/club/) | Powered by Discuz! X2.5 |