← all repositories

ternaus/retinaface

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

412 stars Python Computer Vision
retinaface
Velocity · 7d
+0.2
★ / day
Trend
steady
star history

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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.