mirror of https://github.com/menyifang/DCT-Net
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
731 B
731 B
MTCNN
pytorch
implementation of inference stage of face detection algorithm described in
Joint Face Detection and Alignment using Multi-task Cascaded Convolutional Networks.
Example
How to use it
Just download the repository and then do this
from src import detect_faces
from PIL import Image
image = Image.open('image.jpg')
bounding_boxes, landmarks = detect_faces(image)
For examples see test_on_images.ipynb
.
Requirements
- pytorch 0.2
- Pillow, numpy
Credit
This implementation is heavily inspired by: