ultra-sonic / mxnet_mtcnn_face_detection

MTCNN face detection

MTCNN_face_detection_and_alignment

About

This is a python/mxnet implementation of Zhang's work . it's fast and accurate, see link.

It should have almost the same output with the original work, for mxnet fans and those can't afford matlab :)

中文blog

Requirement

• opencv

  ​ I use cv2 for image io and resize(much faster than skimage), the input image's channel is acutally BGR

• mxnet

  ​ please update to the newest version, we need 'full' mode in Pooling operation

Only tested on Linux and Mac

Test

run:

python main.py

you can change ctx to mx.gpu(0) for faster detection

--- update 20161028 ---

by setting num_worker=4 accurate_landmark=False we can reduce the detection time by 1/4-1/3, the bboxes are still the same, but we skip the last landmark fine-tune stage( mtcnn_v1 ).

--- update 20161207 ---

add function extract_face_chips, examples:

see mtcnn_detector.py for the details about the parameters. this function use dlib's align strategy, which works well on profile images :)

Results

License

MIT LICENSE

Reference

K. Zhang and Z. Zhang and Z. Li and Y. Qiao Joint, Face Detection and Alignment Using Multitask Cascaded Convolutional Networks, IEEE Signal Processing Letters