chexenia / pycaffe-mtcnn

MTCNN with pycaffe

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

MTCNN with Caffe in Python

Usage

python main.py [image_path] [GPU_ID]

e.g.

# Use CPU.
python main.py ./test/1.jpg

# Use GPU 0.
python main.py ./test/1.jpg 0

Test

1.jpg
2.jpg
3.jpg

It takes:

./test/1.jpg  49ms  
./test/2.jpg  99ms  
./test/3.jpg  140ms  

on a Nvidia K40m GPU.

If you look for faster code, check MTCNN in C++.

About

MTCNN with pycaffe


Languages

Language:Python 100.0%