Skip to content

Commit

Permalink
fix errors, refactor, add docs
Browse files Browse the repository at this point in the history
  • Loading branch information
TropComplique committed Nov 4, 2017
1 parent 06173bb commit 440620b
Show file tree
Hide file tree
Showing 16 changed files with 479 additions and 561 deletions.
26 changes: 25 additions & 1 deletion README.md
@@ -1 +1,25 @@
# mtcnn-pytorch
# MTCNN

`pytorch` implementation of **inference stage** of face detection algorithm described in [Joint Face Detection and Alignment using Multi-task Cascaded Convolutional Networks](https://arxiv.org/abs/1604.02878).

## Example
![example of a face detection](images/example.png)

## How to use it
Just download the repository and then do this
```python
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:
* [pangyupo/mxnet_mtcnn_face_detection](https://github.com/pangyupo/mxnet_mtcnn_face_detection)
122 changes: 0 additions & 122 deletions detector.py

This file was deleted.

212 changes: 0 additions & 212 deletions extract_weights_from_caffe_models.ipynb

This file was deleted.

0 comments on commit 440620b

Please sign in to comment.