This repo contains face recognition script written with Pytorch. Tha algorithm is based on the pretrained Inception Res V1 model, details here. I also wrote a story on Medium.
The main concept are:
- Generates face samples
- Load dataset and train only the last linear layer of Inception Res V1
- Test model and visualize examples from validations set