You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
git clone https://github.com/zhenglab/spiralnet.git
cd spiralnet
Install PyTorch and 1.0+ and other dependencies (e.g., torchvision).
For pip users, please type the command pip install -r requirement.txt.
For Conda users, you can create a new Conda environment using conda env create -f environment.yaml.
ImagineGAN
Training
python train.py --path=$configpath$
For example: python train.py --path=./checkpoints/ImagineGAN/celeba/
Testing
python test.py --path=$configpath$
For example: python test.py --path=./checkpoints/ImagineGAN/celeba/
SliceGAN
Put the ImagineGAN model in the corresponding directory, for example, checkpoints/SliceGAN/celeba/imagine_g.pth.
Training
python train.py --path=$configpath$
For example: python train.py --path=./checkpoints/SliceGAN/celeba/
Testing
python test.py --path=$configpath$
For example: python test.py --path=./checkpoints/SliceGAN/celeba/
Citing
@inproceedings{guo2020spiralnet,
author = {Guo, Dongsheng and Liu, Hongzhi and Zhao, Haoru and Cheng, Yunhao and Song, Qingwei and Gu, Zhaorui and Zheng, Haiyong and Zheng, Bing},
title = {Spiral Generative Network for Image Extrapolation},
booktitle = {The European Conference on Computer Vision (ECCV)},
pages={701--717},
year = {2020}
}
