AvLab-CV / Dual_Generator_Face_Reenactment

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Dual-Generator-Face-Reenactment-DG

Python 3.6 CUDA 10.2 Pytorch 1.6

image image

image image image

Abstract: We propose the Dual-Generator (DG) network for large-pose face reenactment. Given a source face and a reference face as inputs, the DG network can generate an output face that has the same pose and expression as of the reference face, and has the same identity as of the source face. As most approaches do not particularly consider large-pose reenactment, the proposed approach addresses this issue by incorporating a 3D landmark detector into the framework and considering a loss function to capture visible local shape variation across large pose. The DG network consists of two modules, the ID-preserving Shape Generator (IDSG) and the Reenacted Face Generator (RFG). The IDSG encodes the 3D landmarks of the reference face into a reference landmark code, and encodes the source face into a source face code. The reference landmark code and the source face code are concatenated and decoded to a set of target landmarks that exhibits the pose and expression of the reference face and preserves the identity of the source face.

Virtual environment

  • Clone this repo to your desired folder
    git clone https://github.com/Charles8745/Dual_Generator_Face_Reenactment.git
    
  • Move to Dual_Generator_Face_Reenactment folder
    cd Dual_Generator_Face_Reenactment
    
  • Establish a virtual environment
    conda env create -f environment.yml
    conda activate reenactment
    pip install -r requirement.txt
    

Download models

Inference

  • Run the demo.py
    python demo.py
    

Details of implementataion

image image

About


Languages

Language:Python 100.0%