insightface/RetinaFace/README.md

RetinaFace Face Detector

Introduction

RetinaFace is a robust single stage face detector which initially described as an arXiv technical report

Training

1. Download groundtruth labels from baiducloud or dropbox and organise the dataset directory under insightface/RetinaFace/ as follows(images can be downloaded from WiderFace website directly):

  data/retinaface/
    train/
      images/
      label.txt
    val/
      images/
      label.txt
    test/
      images/
      label.txt

Testing

Please check test.py for model usage.

Pretrained Model: RetinaFace-R50 is a medium size model with ResNet50 backbone. WiderFace validation mAP: Easy 96.5, Medium 95.6, Hard 90.4. It can output face bounding boxes and five landmarks in a single forward pass.