Awesome
This is the code repo for Facial Details Synthesis From Single Input Image. [Paper] [Supplemental Material] [Video]
This repository consists of 5 individual parts: DFDN, emotionNet, landmarkDetector, proxyEstimator and faceRender.
- DFDN is used to estimate displacement map, and its network architecture is based on junyanz's pix2pix
- For landmarkDetector and FACS-based expression detector (you can choose between this and emotionNet), we use a simplified version of openFace
- proxyEstimator is used to generate proxy mesh using expression/emotion prior. It is modified based on patrikhuber's fantastic work eos
- faceRender is used for interactive rendering
We would like to thank each of the related projects for their great work.
Facial Details Synthesis
Anpei Chen, Zhang Chen, Guli Zhang, Ziheng Zhang, Kenny Mitchell, Jingyi Yu
We present a single-image 3D face synthesis technique that can handle challenging facial expressions while recovering fine geometric details. Our technique employs expression analysis for proxy face geometry generation and combines supervised and unsupervised learning for facial detail synthesis. On proxy generation, we conduct emotion prediction to determine a new expression-informed proxy. On detail synthesis, we present a Deep Facial Detail Net (DFDN) based on Conditional Generative Adversarial Net (CGAN) that employs both geometry and appearance loss functions. For geometry, we capture 366 high-quality 3D scans from 122 different subjects under 3 facial expressions. For appearance, we use additional 163K in-the-wild face images and apply image-based rendering to accommodate lighting variations. Comprehensive experiments demonstrate that our framework can produce high-quality 3D faces with realistic details under challenging facial expressions.
Features
- Functionality
- Proxy estimation with expression/emotion prior
- Facial details prediction, i.e. winkles
- Renderer for results (proxy mesh + normalMap/displacementMap)
- Input: Single image or image folder
- Output: Proxy mesh & texture, detailed displacementMap and normalMap
- OS: Windows 10
Set up environment
- Install windows version of Anaconda Python3.7 and pytorch
- [Optional] Install tensorflow and keras if you want to use emotion prior (emotionNet)
Released version
-
Download the released package.
Release v0.1.0 [Google Drive, OneDrive]
-
Download models and pre-trained weights.
DFDN checkpoints [Google Drive, OneDrive] unzip to
./DFDN/checkpoints
landmark models [Google Drive, OneDrive] unzip to
./landmarkDetector
[Optional] emotionNet checkpoints [Google Drive, OneDrive] unzip to
./emotionNet/checkpoints
-
Install BFM2017
-
Install eos-py by
pip install --force-reinstall eos-py==0.16.1
-
Download BFM2017 and copy
model2017-1_bfm_nomouth.h5
to./proxyEstimator/bfm2017/
-
Run
python convert-bfm2017-to-eos.py
to generatebfm2017-1_bfm_nomouth.bin
in./proxyEstimator/bfm2017/
folder
-
-
Have fun!
Usage
-
For proxy estimation,
python proxyPredictor.py -i path/to/input/image -o path/to/output/folder [--FAC 1][--emotion 1]
-
For batch processing, you can set
-i
to a image folder. -
For prior features, you can optional choose one of those two priors: FACS-based expression prior,
--FAC 1
, emotion prior,--emotion 1
.
example:
python proxyPredictor.py -i ./samples/proxy -o ./results
-
-
For facial details estimation,
python facialDetails.py -i path/to/input/image -o path/to/output/folder
example:
python facialDetails.py -i ./samples/details/019615.jpg -o ./results
python facialDetails.py -i ./samples/details -o ./results
- note: we highly suggest you crop input image to a square size.
Compiling
We suggest you directly download the released package for convenience. If you are interested in compiling the source code, please go through the following guidelines.
-
First, clone the source code,
git clone https://github.com/apchenstu/Facial_Details_Synthesis.git --recursive
-
cd to the root of each individual model then start compiling,
landmarkDetector
-
Executing the
download_libraries.ps1
anddownload_models.ps1
with PowerShell script. -
Open
OpenFace.sln
using Visual Studio and compile the code.After compiling, the excuse file would located in
/x64/Release/FaceLandmarkImg.exe
textureRender
-
install with
mkdir build && cd build cmake -A X64 -D CMAKE_PREFIX_PATH=../thirds ../src
-
Open
textureRender.sln
using Visual Studio and compile the code.After compiling, the excuse file would located in
Release/textureRender.exe
proxyEstimator
-
install vcpkg
-
install package under vcpkg folder:
./vcpkg install opencv boost --triplet x64-windows
-
Install with,
mkdir build && cd build cmake .. -A X64 -DCMAKE_TOOLCHAIN_FILE=[vcpkg root]\scripts\buildsystems\vcpkg.cmake
-
Open
eos.sln
using Visual Studio and compile the code.After compiling, the excuse file would located in
Release/eso.exe
For more details, please refer to this repo.
faceRender
-
Install with
mkdir build && cd build cmake -A X64 -D CMAKE_PREFIX_PATH=../thirds ../src
-
Open
hmrenderer.sln
using Visual Studio and compile the code.After compiling, the excuse file would located in
build\Release
Note: The visualizer currently only supports mesh + normalMap, but will also support displacementMap in the near future.
After compiling, please download DFDN checkpoints, unzip to
./DFDN/checkpoints
. Then you are free to use. -
Others
On the way .....
Q & A
-
Proxy result is different with showing in the paper?
It's because the released version are using a lower resolution input and a different expression dictionary, which are more robust in general case. Please try this if you want to obtain similar results as in the paper.
Citation
If you find this code useful to your research, please consider citing:
@inproceedings{chen2019photo,
title={Photo-Realistic Facial Details Synthesis from Single Image},
author={Chen, Anpei and Chen, Zhang and Zhang, Guli and Mitchell, Kenny and Yu, Jingyi},
booktitle={Proceedings of the IEEE International Conference on Computer Vision},
pages={9429--9439},
year={2019}
}