Skip to content

LiYuhangUSTC/Sketch2Face

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

22 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Sketch2Face

This is the official implementation of paper DeepFacePencil: Creating Face Images from Freehand Sketches. arXiv, Project

Also see another project: Lines2Face project

architecture

Prerequisites

PyTorch 1.6, Python 3.7, NumPy, scipy, PIL, tqdm

Data

We use CelebAMask-HQ to obtain synthesized (boundary) sketches. Deformed sketches are generated by vectorizing synthesized sketches using AutoTrace and adding random offsets to endpoints and control points of vectorized strokes.

Pretrained model

GoogleDrive

Test

The test sketches should be put in folder ./datasets/CelebAMask/test_A. The pretrained model should be put in folder ./checkpoints/pretrained. Example script of testing can be found in ./test_scripts.sh. The results are supposed to be in ./results. The deform.pth file is uploaded to Google Drive.

Cite

@inproceedings{DeepFacePencil,
 author = {Li, Yuhang and Chen, Xuejin and Yang, Binxin and Chen, Zihan and Cheng, Zhihua and Zha, Zheng-Jun},
 title = {DeepFacePencil: Creating Face Images from Freehand Sketches},
 booktitle = {Proceedings of the 28th ACM International Conference on Multimedia},
 series = {MM '20},
 year = {2020},
 isbn = {978-1-4503-7988-5/20/10},
 location = {Seattle, WA, USA},
 pages = {},
 numpages = {9},
 url = {http://doi.acm.org/10.1145/3394171.3413684},
 doi = {10.1145/3394171.3413684},
 acmid = {3413684},
 publisher = {ACM},
 address = {New York, NY, USA},
 keywords = {Sketch-based synthesis, face image generation, spatial attention, dual generator, conditional generative adversarial networks},
} 

Credits

This code borrows heavily from pix2pixHD and pytorch-CycleGAN-and-pix2pix.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published