Skip to content

pytorch implementation of face detection and cropping algorithm MTCNN

License

Notifications You must be signed in to change notification settings

apachetechnology/wsMTCNN-pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

mtcnn-pytorch

pytorch implementation of face detection algorithm MTCNN

Usage MTCNN

Just download the repository and then do this

from model.FaceDetector import CFaceDetector
from model.utils import show_bboxes
from PIL import Image

image = Image.open('images/test3.jpg')

obj = CFaceDetector()
bounding_boxes, landmarks = obj.detect_faces(image)
image = show_bboxes(image, bounding_boxes, landmarks)
image.show()

Requirements

  • pytorch 0.4
  • Pillow, numpy

Credit

Reference

MTCNN: Joint Face Detection and Alignment using Multi-task Cascaded Convolutional Networks.

About

pytorch implementation of face detection and cropping algorithm MTCNN

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published