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

StanislasBertrand
195 Stars 19 Forks MIT License 49 Commits 2 Opened issues

Description

RetinaFace (RetinaFace: Single-stage Dense Face Localisation in the Wild, published in 2019) reimplemented in Tensorflow 2.0, with pretrained weights available !

Services available

!
?

Need anything else?

Contributors list

# 171,981
Keras
C
C++
insight...
18 commits
# 97,366
C
insight...
retinaf...
tensorf...
2 commits

RetinaFace-tf2

RetinaFace (RetinaFace: Single-stage Dense Face Localisation in the Wild, published in 2019) reimplemented in Tensorflow 2.0, with pretrained weights available. Resnet50 backbone.

Original paper -> arXiv
Original Mxnet implementation -> Insightface

Table of contents

  1. Installation
  2. Usage
  3. Benchmark
  4. Evaluation
  5. Acknowledgements

example output : testing on a random internet selfie


INSTALLATION

To install dependencies, run :

pip install -r requirements.txt

Then build the rcnn module by running :

make

USAGE

Run :

angular2
python detect.py --sample_img="./sample-images/WC_FR.jpeg"
Python usage : ```python from src.retinafacetf2.retinaface import RetinaFace import cv2

detector = RetinaFace(False, 0.4) img = cv2.imread("./sample-images/WC_FR.jpeg") faces, landmarks = detector.detect(img, 0.9) ```

BENCHMARK

mAP result values on the WIDERFACE validation dataset:

| Model | Easy | Medium | Hard | |---|---|---|---| |Original Mxnet implementation | 96.5 | 95.6 | 90.4 | | Ours | 95.6 | 94.6 | 88.5 |

EVALUATE ON WIDERFACE

In order to verify the models accuracy on the WiderFace dataset: * Run the model on the dataset and generate text files as results

angular2
python eval_widerface --widerface_data_dir = "/data/WIDER_test/images" --save_folder="./WiderFace-Evaluation/results/"
* Evaluate the results
angular2
cd ./WiderFace-Evaluation
python setup.py build_ext --inplace
python evaluation.py -p ./results_val/ -g ./ground_truth/

ACKNOWLEDGEMENTS

This work is largely based on the original implementation by the amazing insightface team
Evaluation on widerface done with the Widerface-Evaluation repo
If you use this repo, please reference the original work :

@inproceedings{Deng2020CVPR,
title = {RetinaFace: Single-Shot Multi-Level Face Localisation in the Wild},
author = {Deng, Jiankang and Guo, Jia and Ververas, Evangelos and Kotsia, Irene and Zafeiriou, Stefanos},
booktitle = {CVPR},
year = {2020}
}

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.