ternaus/retinaface
A PyTorch implementation of RetinaFace for face detection with pre-trained models, distributed training, and mAP evaluation.

This repository is a PyTorch implementation of RetinaFace, a face detection deep learning model. It uses PyTorch Lightning for the training loop, providing distributed training, fp16 support, and synchronized BatchNorm across GPUs. The project includes pre-trained models for inference, uses Albumentations for data augmentation, and computes mAP metrics during validation. Models can be installed via pip and used for face detection in images.