insightface/alignment/synthetics/README.md

Introduction

We provide training and testing tools on synthetics data.

Dataset

Training dataset

Download Face Synthetics dataset from https://github.com/microsoft/FaceSynthetics and put it somewhere.

Then use tools/prepare_synthetics.py for training data preparation.

Testing dataset

300-W

Pretrained Model

ResNet50d

Train and Test

Prerequisites

• pytorch_lightning
• timm
• albumentations

Training

python -u trainer_synthetics.py

which uses resnet50d as backbone by default, please check the code for detail.

Testing

Please check test_synthetics.py for detail.

Result Visualization(3D 68 Keypoints)