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 :)
-
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
run:
python main.py
you can change ctx to mx.gpu(0) for faster detection
--- update ---
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 ---
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 :)
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
Reference: https://github.com/YYuanAnyVision/mxnet_mtcnn_face_detection