Pytorch implementation of the Face R-FCN deep learning architecture for detecting faces - https://arxiv.org/abs/1709.05256 This was an attempt to completely segregate each component and write cleaner code to help others understand the individual steps behind the complex Face R-FCN architecture.
Versions required: Pytorch 0.4.1 and CUDA 9.2
- Compile CUDA modules: cd models && sh make.sh
- For training: python3 training.py
- For inference/validation: python3 inference.py