Skip to content

Z-Jeff/Cat_Face_Detection

Folders and files

NameName
Last commit message
Last commit date

Latest commit

author
118785
Jan 31, 2020
3425482 · Jan 31, 2020

History

19 Commits
Jan 31, 2020
Jan 31, 2020
Jan 31, 2020
Jan 31, 2020
Jan 31, 2020
Jan 31, 2020
Jan 31, 2020

Repository files navigation

Cat face detection using MTCNN

results:

  • It doesn't perform well, only for reference

Test an image

  • run > python mtcnn_test.py

Training data prepraring

  • download cat face dataset for landmark, then unzip it into ./data_set/original/
    • run > python ./anno_store/tool/transform.py change train.csv into .txt(anno_train.txt)

Training

  • preparing data for P-Net

    • run > python mtcnn/data_preprocessing/gen_Pnet_train_data.py
    • run > python mtcnn/data_preprocessing/assemble_pnet_imglist.py
  • train P-Net

    • run > python mtcnn/train_net/train_p_net.py
  • preparing data for R-Net

    • run > python mtcnn/data_preprocessing/gen_Rnet_train_data.py (maybe you should change the pnet model path)
    • run > python mtcnn/data_preprocessing/assemble_rnet_imglist.py
  • train R-Net

    • run > python mtcnn/train_net/train_r_net.py
  • preparing data for O-Net

    • run > python mtcnn/data_preprocessing/gen_Onet_train_data.py
    • run > python mtcnn/data_preprocessing/gen_landmark_48.py
    • run > python mtcnn/data_preprocessing/assemble_onet_imglist.py
  • train O-Net

    • run > python mtcnn/train_net/train_o_net.py

Citation

DFace mtcnn-pytorch

About

Cat face detection

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages