dualFace: Two-Stage Drawing Guidance for Freehand Portrait Sketching

dualFace

dualFace: Two-Stage Drawing Guidance for Freehand Portrait Sketching (CVMJ)

We provide python implementations for our CVM 2021 paper "dualFace:Two-Stage Drawing Guidance for Freehand Portrait Sketching". This project provide sketch support for artistic portrait drawings with a two-stage framework.

User Interface

Network Structure (Global Stage)

dualface-global

Network Structure (Local Stage)

dualface-local

Prerequisites

Window
Conda (Python 3.6)
CPU or NVIDIA GPU + CUDA CuDNN

Getting Started

Installation

Install PyTorch 1.3.1 and torchvision 0.4.1 from http://pytorch.org and other dependencies (e.g., visdom and dominate). You can install all the dependencies by

bat
call conda remove -n py36df
call conda create -n py36df python=3.6 
call conda activate py36df
call conda install pytorch==1.3.1 -c pytorch
pip install cmake
pip install -r requirements.txt

Quick Start (Apply a Pre-trained Model)

Download a pre-trained model from (https://drive.google.com/open?id=1cQx9hPOJ18sU5HPGkbRTJ-e6cYqqHUHh)

cd sse
sse.exe "-i index_file -v vocabulary -f filelist -n 8"
call conda activate py36df
python demo.py

Acknowledgments

Our code has depended on the following opensource codes.

MaskGAN(https://github.com/switchablenorms/CelebAMask-HQ)
faceParsing(https://github.com/zllrunning/face-parsing.PyTorch)
APDrawingGAN(https://github.com/yiranran/APDrawingGAN)
OpenSSE(https://github.com/zddhub/opensse)

Please contact [email protected] for any comments or requests.

Citation

If you use this code for your research, please cite our paper.

@article{huang21dualface,
  title={dualFace: Two-Stage Drawing Guidance for Freehand Portrait Sketching},
  author={Huang, Zhengyu and Peng, Yichen and Hibino, Tomohiro and Zhao, Chunqi Zhao and Xie, Haoran and Fukusato, Tsukasa and Miyata, Kazunori},
  journal={Computational Visual Media},
  year={2021},
  publisher={Springer}
}