PaddlePaddle-MTCNN

基于PaddlePaddle复现的MTCNN人脸检测模型

APACHE-2.0 License

Stars
25

MTCNNMulti-task convolutional neural networkP-NetR-NetO-Net2016P-NetR-NetO-Net

  • PaddlePaddle 2.0.1
  • Python 3.7

  • models/Loss.py MTCNN
  • models/PNet.py PNet
  • models/RNet.py RNet
  • models/ONet.py ONet
  • utils/data_format_converter.py
  • utils/data.py
  • utils/utils.py
  • train_PNet/generate_PNet_data.py PNet
  • train_PNet/train_PNet.py PNet
  • train_RNet/generate_RNet_data.py RNet
  • train_RNet/train_RNet.py RNet
  • train_ONet/generate_ONet_data.py ONet
  • train_ONet/train_ONet.py ONet
  • infer_path.py
  • infer_camera.py

PNetRNetONet

PNet

PNetProposal NetworkP-Net

  • cd train_PNet train_PNet
  • python3 generate_PNet_data.py PNet
  • python3 train_PNet.py PNet

RNet

Refine NetworkP-NetP-NetR-NetBounding-Box RegressionNMS

  • cd train_RNet train_RNet
  • python3 generate_RNet_data.py PNetRNet
  • python3 train_RNet.py RNet

ONet

ONetOutput NetworkR-NetO-NetR-Net

  • cd train_ONet train_ONet
  • python3 generate_ONet_data.py PNetRNetONet
  • python3 train_ONet.py ONet

  • python3 infer_path.py box

  • python3 infer_camera.py box

  1. https://github.com/AITTSMD/MTCNN-Tensorflow
  2. https://blog.csdn.net/qq_36782182/article/details/83624357
Related Projects