Skip to content

Files

Latest commit

f2b4c3d · Jan 9, 2022

History

History

synthetics

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)