search

XgTu / 2DASL

The code (pytorch for testing & matlab for 3D plot and evaluation) for our project: Joint 3D Face Reconstruction and Dense Face Alignment from A Single Image with 2D-Assisted Self-Supervised Learning （2DASL）
MIT License
467 stars 116 forks source link

readme

Our 2D-Assisted Self-supervised Learning (2DASL) is available at: https://arxiv.org/abs/1903.09359

This project is created by Tu Xiaoguang (xguangtu@outlook.com) and Luo Yao (luoyao_alpha@outlook.com). Any questions pls open issues for our project, we will reply quickly.

To facility the research in the community of 3D face reconstruction and 3D face alignment, we release our source code, including the pytorch code for testing (training code will be available upon the acceptance of our paper), the matlab code for 3D plot, 3D face rendering and evaluation.

Landmark detection (left: only 68 landmarks are plotted to show), 3D face reconstruction (middle) and Dense face alignment (right)

Face swapping

Facial expression retargeting

Visual results

3D face reconstruction & face alignment

Comparison with AFLW2000-3D groundth (Green: landmarks predicted by our 2DASL. Red: ground truth of AFLW2000-3D)

Evaluation results (face alignment)

Performance comparison on AFLW2000-3D (68 2D landmarks) and AFLW-LFPA (34 2D visible landmarks). The NME (%) for faces with different yaw angles are reported.

Error Distribution Curves (EDC) of face alignment results on AFLW2000-3D.

Evaluation results (3d face reconstruction)

Comparisons on AFLW2000-3D

We add the matlab evaluation code with the metric Normalized Mean Error (NME) in the folder "evaluation", including:

  1. We compare the results of our 2DASL with PRNet on the sparse 68 key points on both 2D&3D coordinates.
  2. We compare the results of our 2DASL with 3DDFA on all points for the task of dense alignment (both 2D and 3D coordinates) and 3D face recnostruction.

  3. The Iterative Closest Point (ICP) algorithm, which is used to find the nearest points between two 3D models is included in our code.

    Usage: Download the "visualize.rar" package at： https://pan.baidu.com/s/1jazeore9M_4khk50Dz_Jlg using the password: "o7ji".

We add our test code in the folders "models" and "test_codes", including:

  1. The 2DASL models for stage1 and stage2, in the folder "models".
  2. The test code to obtain 3D vertex for ALFW2000-3D images.

  3. The code to plot 2D facial landmarks.

    Usage: python-3.6.6, pytorch-0.4.1. Just run the script "main_visualize_plot_4chls.py" in the folder "test_codes".

We add our 3D face rendering code in the folder "3D_results_plot", including:

  1. The code for 3D face rendering, i.e., drawing 3d reconstruction mask.
  2. The code for plotting dense face alignment.

  3. The code for writing .obj file, the .obj files could be opend by the software meshlab.

    Usage: Download the "visualize.rar" package at： https://pan.baidu.com/s/1M0sg3eqrtzxxMJqmAzczRw using the password: "h26t". Or download the package here: https://drive.google.com/file/d/1414VE7oiusKxeyx0LYZOv042c8DfhFPf/view?usp=sharing Putting the package under the folder "3D_results_plot", extracting it and then run the script "test_good_byXgtu.m".

Acknowledgement

Thanks for the authors of 3DDFA and PRNet for making their excellent works publicly available.

Additional example

// xgtu-lmks_res

Citation

If you find our code is useful for your research, pls cite our work:

@article{tu2019joint,
  title={Joint 3D Face Reconstruction and Dense Face Alignment from A Single Image with 2D-Assisted Self-Supervised Learning},
  author={Tu, Xiaoguang and Zhao, Jian and Jiang, Zihang and Luo, Yao and Xie, Mei and Zhao, Yang and He, Linxiao and Ma, Zheng and Feng, Jiashi},
  journal={arXiv preprint arXiv:1903.09359},
  year={2019}
}