Skip to main content

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 & hash File type Python version Upload date
torch_mtcnn-0.0.7-py3-none-any.whl (9.6 kB) View hashes Wheel py3
torch-mtcnn-0.0.7.tar.gz (7.2 kB) View hashes Source None

Supported by

Elastic Elastic Search Pingdom Pingdom Monitoring Google Google BigQuery Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN SignalFx SignalFx Supporter DigiCert DigiCert EV certificate StatusPage StatusPage Status page