MTCNN-with-Caffe-in-C++
A implement to MTCNN.
Joint Face Detection and Alignment using Multi-task Cascaded Convolutional Neural Networks
paper url: spl.pdf
Requirement
- Caffe:https://github.com/BVLC/caffe, 中文版目前坑最少的编译环境配置帮助文档:https://blog.csdn.net/fengtaoO08/article/details/83023428
有问题欢迎提在issues里。
- VS2017:https://visualstudio.microsoft.com/zh-hans/thank-you-downloading-visual-studio/?sku=Community&rel=15
- library maybe inneed:opencv3.1,OpenBLAS-0.3.6
- dataset:widerface,celebA
Result
配置vs2017
需要配置一些依赖项
属性页
- 常规-》平台工具集(v141)
- 调试-》包含目录、库目录
- C/C++-》常规
- 链接器-》输入,配置静态库
训练
将wilderFace数据集解压放到RaiseData文件夹下,UNcomment掉12的netsize,运行生产12x12的样本图片,然后24、48.
值得注意的是,48net的FP率会偏高,需要生产hardexample来强化学习,生成这个样板的脚本是gen_hard_example.py.
最后生成imdb格式数据集就可以训练了。
我的训练网络提供在RaiseData文件夹下,需要用得话记得修改路径。
运行
运行Myfacedetection_mtcnn.cpp,选择图片输入,进过卷积网络后得到输出。