Skip to main content
Join the official 2020 Python Developers SurveyStart the survey!

Implementation of MTCNN using Pytorch.

Project description

The original work of this project is always belong to the original creator (https://github.com/TropComplique/mtcnn-pytorch). I just make this available on pypi for easy installation. To install this project just type pip install torch-mtcnn

MTCNN

pytorch implementation of inference stage of face detection algorithm described in
Joint Face Detection and Alignment using Multi-task Cascaded Convolutional Networks.

Example

example of a face detection

How to use it

Install the package with pip: pip install torch-mtcnn

from torch_mtcnn import detect_faces
from PIL import Image

image = Image.open('image.jpg')
bounding_boxes, landmarks = detect_faces(image)

For a few more examples available on the original repository (link above).

Requirements

  • pytorch 0.2
  • Pillow, numpy

Credit

This implementation is heavily inspired by:

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Files for torch-mtcnn, version 0.0.7
Filename, size File type Python version Upload date Hashes
Filename, size torch_mtcnn-0.0.7-py3-none-any.whl (9.6 kB) File type Wheel Python version py3 Upload date Hashes View
Filename, size torch-mtcnn-0.0.7.tar.gz (7.2 kB) File type Source Python version None Upload date Hashes View

Supported by

Pingdom Pingdom Monitoring Google Google Object Storage and Download Analytics Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN DigiCert DigiCert EV certificate StatusPage StatusPage Status page