Qiulin-W / SAFA

Official Pytorch Implementation of 3DV2021 paper: SAFA: Structure Aware Face Animation.
Other
177 stars 29 forks source link

Checkpoints of the 3DMM pre-trained model #23

Open HowieMa opened 1 year ago

HowieMa commented 1 year ago

Dear authors, Thanks for sharing such a wonderful and interesting project! May I ask if you could release the pre-trained 3DMM model? i.e., the checkpoint after running the 3DMM Estimator Pre-training step with code

CUDA_VISIBLE_DEVICES="0,1,2,3" python -m torch.distributed.launch --nproc_per_node 4 run_ddp.py --config config/pretrain.yaml

As I'd like to run your end-to-end training step with a fully pre-trained 3DMM model given by "--tdmm_checkpoint".

Looking forward to your reply. Thanks in advance!