Skip to content

Face Texture Inpainting with Score-Based Generative Models

Notifications You must be signed in to change notification settings

ardarslan/score-face

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

40 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Score-Face

Go to the main folder of this repository. Create and activate its conda environment. Prepare its data.

cd score-face
conda env create -f environment.yml
conda activate score-face
mkdir assets
cd assets
gdown --id 1sXrlgTC6U2jzWCIUZTbRcY3AbOJfcbbu
gdown --id 1-mtdSwuefIZA0n85QWScQo2WRvJNWwUy

Run the code.

Go into src folder.

cd ../src

To do optimization in the image space:

python3 main.py --input_obj_path INPUT_OBJ_PATH --optimization_space image --order_views true --two_rounds true --num_corrector_steps 1 --snr 0.15

To do optimization in the texture space:

python3 main.py --input_obj_path INPUT_OBJ_PATH --optimization_space texture --order_views true --two_rounds true --num_corrector_steps 6 --snr 0.015

Results

Optimization in the Image Space

unoptimized_animation optimized_animation

unoptimized_animation optimized_animation

unoptimized_animation optimized_animation

ezgif-3-3d001e314f ezgif-3-bbc31152a3

Optimization in the Texture Space

unoptimized_animation optimized_animation

unoptimized_animation optimized_animation

unoptimized_animation optimized_animation

unoptimized_animation optimized_animation

Note: The code under the directory https://github.com/ardarslan/score-face/tree/master/src/score_sde was adapted from https://github.com/yang-song/score_sde_pytorch.

About

Face Texture Inpainting with Score-Based Generative Models

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published