Face API with Docker, Tensorflow and PyTorch
- Face Detection - TensorFlow (SSD-Mobilenet)
- Facial Emotion Recognition - PyTorch (Resnet50)
- Age and Gender Estimation - Keras (WideResnet)
- Facial Recognition - PyTorch (VGGFace2)
- Docker with Nvidia Support Nvidia-Docker
- CUDA 9 Guide
Make sure the models and images are inside their respective folders.
Run all the docker syncronically:
bash runmodels.sh
Stop docker containers:
bash stopmodels.sh
Test that the models are running using:
python test_api.py