[Paper] [Project Page] [Interactive Demo] [Supplementary Material]
The code only has been tested on Ubuntu 20.04.
GPU is required.
- Clone the repo:
git clone --single-branch https://github.com/zengxianyu/sketchedit
git submodule init
git submodule update
- Install using the provided docker file
Dockerfile
or
conda env create -f environment.yml
- Switch to the installed environment
conda activate editline
- Download pretrained model
chmod +x download/*
./download/download_model.sh
- Use the model trained on CelebAHQ
./test_celeb.sh
- Use the model trained on Places
./test_places.sh
Training code and data will be released after the paper is published.