Self-Supervised Monocular 3D Face Reconstruction by Occlusion-Aware Multi-view Geometry Consistency(ECCV 2020)

This is an official python implementation of MGCNet. This is the pre-print version https://arxiv.org/abs/2007.12494.

Result

1. video

   

2. image

3. Full video can be seen in [YouTube] https://www.youtube.com/watch?v=DXzkO3OwlYQ

Running code

1. Code + Requirement + thirdlib

We run the code with python3.7, tensorflow 1.13

bash
git clone --recursive https://github.com/jiaxiangshang/MGCNet.git
cd MGCNet
(sudo) pip install -r requirement.txt

(2) Then how to compile tfmeshrender, my setting is bazel==10.1, gcc==5., the compile command is

bash
bazel build ...

2.Model

1. 3dmm model + network weight

We include BFM09/BFM09 expression, BFM09 face region from https://github.com/microsoft/Deep3DFaceReconstruction, BFM09 uv from https://github.com/anilbas/3DMMasSTN into a whole 3dmm model. https://drive.google.com/file/d/1RkTgcSGNs2VglHriDnyr6ZS5pbnZrUnV/view?usp=sharing Extract this file to /MGCNet/model 2. pretain

This include the pretrail model for the Resnet50 and vgg pretrain model for Facenet. https://drive.google.com/file/d/1jVlf05BmnbIQXZRfmz-dA03xGCawBw/view?usp=sharing Extract this file to /MGCNet/pretain

3.Data

1. data demo: https://drive.google.com/file/d/1Du3iRO0GNncZsbK4K5sboSeCUv0-SnRV/view?usp=sharing

Extract this file to /MGCNet/data, we can not provide all datas, as it is too large and the license of MPIE dataset http://www.cs.cmu.edu/afs/cs/project/PIE/MultiPie/Multi-Pie/Home.html not allow me to do this.

1. data: landmark ground truth

The detection method from https://github.com/1adrianb/2D-and-3D-face-alignment, and we use the SFD face detector

1. data: skin probability

I get this part code from Yu DENG([email protected]), maybe you can ask help from him.

4.Testing

1. test_image.py This is used to inference a single unprocessed image(cmd in file). This file can also render the images(geometry, texture, shading,multi-pose), like above or in our paper(read code), which makes visualization and comparison more convenient.

2. preprocess All the preprocess has been included in 'testimage.py', we show the outline here. (1) face detection and face alignment are package in ./tools/preprocess/detectlandmark,py. (2) face alignment by affine transformation to warp the unprocess image. Test all the images in a folder can follow this preprocess.

5.Training

1. train_unsupervise.py

Useful tools(keep updating)

1. face alignment tools
2. 3D face render tools.
3. Camera augment for rendering.

Citation

If you use this code, please consider citing:

@article{shang2020self,
  title={Self-Supervised Monocular 3D Face Reconstruction by Occlusion-Aware Multi-view Geometry Consistency},
  author={Shang, Jiaxiang and Shen, Tianwei and Li, Shiwei and Zhou, Lei and Zhen, Mingmin and Fang, Tian and Quan, Long},
  journal={arXiv preprint arXiv:2007.12494},
  year={2020}
}

Contacts

Please contact [email protected] or open an issue for any questions or suggestions.

Acknowledgements