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
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 examples see test_on_images.ipynb
.
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.
Source Distribution
torch-mtcnn-0.0.5.tar.gz
(7.0 kB
view hashes)
Built Distribution
Close
Hashes for torch_mtcnn-0.0.5-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 1386db20c625b42583917fc5981b564617d5c986e87b1fe04f5de937507a0c01 |
|
MD5 | 76dfcbbab2198540f50fcf91c6245dd0 |
|
BLAKE2b-256 | 3e34ed965b41f1aee7c6c32bc9e19233494c7b5b16581a1798be0e5412dcf54e |