Need help with mtcnn-pytorch?
Click the “chat” button below for chat support from the developer who created it, or find similar developers for support.

About the developer

Sierkinhane
254 Stars 107 Forks 6 Commits 15 Opened issues

Description

A face detection algorithm

Services available

!
?

Need anything else?

Contributors list

# 23,619
Python
machine...
pytorch
ocr
6 commits

mtcnn-pytorch

Descriptions in chinese

https://blog.csdn.net/Sierkinhane/article/details/83308658

results:

Test an image

  • run > python mtcnn_test.py

Training data prepraring

  • download WIDER FACE (passcode:lsl3) face detection data then store it into ./dataset/facedetection
    • run > python ./annostore/tool/format/transform.py change .mat(widerfacetrain.mat) into .txt(annotrain.txt)
  • download CNN_FacePoint face detection and landmark data then store it into ./dataset/facelandmark

Training

  • preparing data for P-Net
    • run > python mtcnn/datapreprocessing/genPnettraindata.py
    • run > python mtcnn/datapreprocessing/assemblepnet_imglist.py
  • train P-Net

    • run > python mtcnn/trainnet/trainp_net.py
  • preparing data for R-Net

    • run > python mtcnn/datapreprocessing/genRnettraindata.py (maybe you should change the pnet model path)
    • run > python mtcnn/datapreprocessing/assemblernet_imglist.py
  • train R-Net

    • run > python mtcnn/trainnet/trainr_net.py
  • preparing data for O-Net

    • run > python mtcnn/datapreprocessing/genOnettraindata.py
    • run > python mtcnn/datapreprocessing/genlandmark_48.py
    • run > python mtcnn/datapreprocessing/assembleonet_imglist.py
  • train O-Net

    • run > python mtcnn/trainnet/traino_net.py

# Citation DFace

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.