0.前言
在这里将MTCNN程序复现一遍,这个Github程序的README是有问题的,程序也有一些问题,我也是仅仅将他的算法训练完毕,由于小白一个,只看了MTCNN的论文,里面有大量的先验论文与知识,对整体的网络实现以及代码还是有些懵懂,算是熟悉了跑模型的流程,其他知识待补充。
复现程序地址:点击打开
1.准备工作
- 需要有与CUDA兼容的GPU来训练模型。
- 应该首先下载WIDER Face和landmark dataset前者用于训练人脸检测,后者用于训练landmark检测。
2.依赖的版本
- Tensorflow 1.2.1
- TF-Slim
- Python 2.7
- Ubuntu 16.04
- Cuda 8.0
3.数据处理以及训练流程
- 下载Wider Face 数据集,将里面的WIDER_train.zip解压到 prepare_data文件并且更名为WIDER_train
- 下载“1.准备工作”中的landmark数据集,将landmark_train.zip数据集解压到prepare_data中
- 运行prepare_data/gen_12net_data