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

Pretrained Model

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)