ElaineBao/mtcnn

Introduction

this repository is the implementation of MTCNN in MXnet

• core: core routines for MTCNN training and testing.
• tools: utilities for training and testing
• data: Refer to Data Folder Structure for dataset reference. Usually dataset contains images and imglists
• model: Folder to save training symbol and model
• prepare_data: scripts for generating training data for pnet, rnet and onet

Useful information

You're required to modify mxnet/src/regression_output-inl.h before using the code for training. cp regression_output-inl.h YOUR_MXNET_ROOT/src/operator/regression_output-inl.h then recompile mxnet make clean && make -j 4 USE_DIST_KVSTORE=1 USE_CUDA=1 USE_CUDA_PATH=/usr/local/cuda USE_CUDNN=1

• Dataset format The images used for training are stored in ./data/dataset_name/images/ The annotation file is placed in ./data/dataset_name/imglists/

  • For training: Each line of the annotation file states a training sample.
    The format is: [path to image] [cls_label] [bbox_label]
    cls_label: 1 for positive, 0 for negative, -1 for part face.
    bbox_label are the offset of x1, y1, x2, y2, calculated by (xgt(ygt) - x(y)) / width(height)
    An example would be 12/positive/28 1 -0.05 0.11 -0.05 -0.11.
    Note that all the strings are seperated by space.

  • For testing: Similar to training but only path-to-image is needed.

• Data Folder Structure (suppose root is data)

cache (created by imdb)
-- name + image set + gt_roidb
-- results (created by detection and evaluation)
mtcnn # contains images and anno for training mtcnn
-- images
---- 12 (images of size 12 x 12, used by pnet)
---- 24 (images of size 24 x 24, used by rnet)
---- 48 (images of size 48 x 48, used by onet)
-- imglists 
---- train_12.txt
---- train_24.txt
---- train_48.txt
custom (datasets for testing) 
-- images
-- imglists
---- image_set.txt

• Scripts to generate training data(from wider face dataset)
  • run wider_annotations/transform.m to get the annotation file of the format we need.
  • gen_pnet_data.py: obtain training samples for pnet
  • gen_hard_example.py: prepare hard examples. you can set test_mode to "pnet" to get training data for rnet, or set test_mode to "rnet" to get training data for onet.
  • gen_imglist.py: ramdom sample images generated by gen_pnet_data.py or gen_hard_example.py to form training set.

Results

License

MIT LICENSE

Reference

Kaipeng Zhang, Zhanpeng Zhang, Zhifeng Li, Yu Qiao , " Joint Face Detection and Alignment using Multi-task Cascaded Convolutional Networks," IEEE Signal Processing Letter