Getting Started

Clone the repo:

    git clone https://github.com/ofnote/3d-face
    cd 3d-face

Requirements

• Python 3.7 (numpy, skimage, scipy, opencv)
• PyTorch >= 1.6 (pytorch3d)
• face-alignment (Optional for detecting face)
  You can run

  pip install -r requirements.txt

  Then follow the instruction to install pytorch3d.

Training

• To generate CSV file containing images paths, first specify the directory inside preprocess.py containing training images Run

  python3 -m decalib.datasets.preprocess

• Specify the dataset foldername inside train file in line 267.
• To start training Run

  python3 -m decalib.trainFromscratch.train

Acknowledgements

• DECA - Used their coarse reconstruction part for reconstructing 3d face.