Need help with retinaface-tf2?
Click the “chat” button below for chat support from the developer who created it, or find similar developers for support.

About the developer

171 Stars 62 Forks MIT License 19 Commits 13 Opened issues


RetinaFace (Single-stage Dense Face Localisation in the Wild, 2019) implemented (ResNet50, MobileNetV2 trained on single GPU) in Tensorflow 2.0+. This is an unofficial implementation. With Colab.

Services available


Need anything else?

Contributors list

# 126,973
17 commits


Language grade: Python Star Fork License

Open In Colab

:fire: RetinaFace (RetinaFace: Single-stage Dense Face Localisation in the Wild, published in 2019) implemented (ResNet50, MobileNetV2 trained on single GPU) in Tensorflow 2.0+. This is an unofficial implementation. :fire:

RetinaFace presents a robust single-stage face detector, which performs pixel-wise face localisation on various scales of faces by taking advantages of joint extra-supervised and self-supervised multi-task learning(manually annotate five facial landmarks on the WIDER FACE). On the WIDER FACE hard test set, RetinaFace outperforms the state of the art average precision (AP) by 1.1% (achieving AP equal to 91.4%).

Original Paper:   Arxiv

Offical Implementation:   MXNet

:: Results from this reporepository. ::





Create a new python virtual environment by Anaconda or just use pip in your python environment and then clone this repository as following.

Clone this repo

git clone
cd retinaface-tf2


conda env create -f environment.yml
conda activate retinaface-tf2


pip install -r requirements.txt

Data Preparing


Step 1: Download the WIDER FACE dataset images from the download links bellow.

| Dataset Name | Link | |:------------:|:----------:| | WIDER Face Training Images | Google Drive | | WIDER Face Validation Images | Google Drive |

Step 2: Download the Retinaface official annotations (face bounding boxes & five facial landmarks) from the download links bellow.

| Dataset Name | Link | |:------------:|:----------:| | Retinaface Annotations | Google Drive / Dropbox |

Step 3: Extract downloaded files into

. The directory structure should be like bellow.

Step 4: Convert the training images and annotations to tfrecord file with the the script bellow. ```bash

Binary Image (recommend): need additional space

python data/ --outputpath="./data/widerfacetrainbin.tfrecord" --isbinary=True

Online Image Loading:

python data/ --outputpath="./data/widerfacetrain.tfrecord" --is_binary=False ```

Note: - You can run

python ./
to check if the dataloader work. Please modify the inside setting (in ./ for different situations.

Training and Testing


Config File

You can modify your own dataset path or other settings of model in ./configs/*.yaml for training and testing, which like below.

# general setting
batch_size: 8
input_size: 640
backbone_type: 'ResNet50'  # 'ResNet50', 'MobileNetV2'
sub_name: 'retinaface_res50'

training dataset

dataset_path: './data/widerface_train_bin.tfrecord' dataset_len: 12880 # number of training samples using_bin: True using_flip: True using_distort: True

testing dataset

testing_dataset_path: './data/widerface/val'


out_channel: 256

anchor setting

min_sizes: [[16, 32], [64, 128], [256, 512]] steps: [8, 16, 32] match_thresh: 0.45 ignore_thresh: 0.3 variances: [0.1, 0.2] clip: False

training setting

epoch: 100 init_lr: !!float 1e-2 lr_decay_epoch: [50, 68] lr_rate: 0.1 warmup_epoch: 5 min_lr: !!float 1e-3

weights_decay: !!float 5e-4 momentum: 0.9

pretrain: True

save_steps: 2000

Note: - The

is the name of outputs directory used in checkpoints and logs folder. (make sure of setting it unique to other models) - The
is used to choose the type of training data, which should be according to the data type you created in the Data-Preparing. - The
is the number interval steps of saving checkpoint file.


Train the Retinaface model by yourself, or dowload it from BenchmarkModels. ```bash

train ResNet50 backbone model

python --cfgpath="./configs/retinafaceres50.yaml" --gpu=0

or train MobileNetV2 backbone model

python --cfgpath="./configs/retinafacembv2.yaml" --gpu=0 ```

Note: - The

is used to choose the id of your avaliable GPU devices with
system varaible. - You can visualize the learning rate scheduling by running "
python ./modules/

Testing on WIDER FACE Validation Set

You can download my trained models for testing from Models without training it yourself. And, evaluate the models you got with the corresponding cfg file on the testing dataset.

Step 1: Produce txt results and visualizations from model. ```bash

Test ResNet50 backbone model

python --cfgpath="./configs/retinaface_res50.yaml" --gpu=0


Test ResNet50 backbone model

python --cfgpath="./configs/retinaface_mbv2.yaml" --gpu=0 ```

Note: - The visualizations results would be saved into


Step 2: Evaluate txt results. (Codes come from Here)

cd ./widerface_evaluate
python build_ext --inplace

Detect on Input Image

You can detect on your image by the model. For example, detect on the image from ./data/0Parademarchingband1149.jpg as following.

python --cfg_path="./configs/retinaface_res50.yaml" --img_path="./data/0_Parade_marchingband_1_149.jpg" --down_scale_factor=1.0
# or
python --cfg_path="./configs/retinaface_mbv2.yaml" --img_path="./data/0_Parade_marchingband_1_149.jpg" --down_scale_factor=1.0

Note: - You can down scale your input by the


Demo on Webcam

Demo face detection on your webcam. ```bash python --cfgpath="./configs/retinafaceres50.yaml" --webcam=True --downscalefactor=1.0


python --cfgpath="./configs/retinafacembv2.yaml" --webcam=True --downscalefactor=1.0 ```



WiderFace Validation Set Performance in single scale

| Model Name | Easy | Medium | Hard | |---------------------|--------|--------|--------| | Tensorflow2 MobileNetV2 | 93.23% | 91.74% | 80.01% | | Tensorflow2 ResNet50 | 94.21% | 93.25% | 83.55% | | Pytorch ResNet50 | 95.48% | 94.04% | 84.43% | | Mxnet ResNet50 (Official) | 94.97% | 93.89% | 82.27% |

Note: - Tensorflow2 version (ResNet50/MobileNetV2) in this repository only trained with batch size (8/16) on single GPU. - All models evaluate on original image scale.



| Model Name | Download Link | |---------------------|---------------| | Retinaface ResNet50 | GoogleDrive | | Retinaface MobileNetV2 | GoogleDrive |

Note: - After dowloading these models, extract them into

for restoring. - All training settings of the models can be found in the corresponding ./configs/*.yaml files. - You can see the tensorboard traning logs in here. - Based on the property of the training dataset, all the pre-trained models can only be used for non-commercial applications.



Thanks for these source codes porviding me with knowledges to complete this repository.

  • (Official)
    • Face Analysis Project on MXNet
    • Retinaface get 80.99% in widerface hard val using mobilenet0.25.
    • Python Evaluation Code for Wider Face Dataset
    • YoloV3 Implemented in TensorFlow 2.0

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.