zhenglab / spiralnet

14 stars 1 forks source link

SpiralNet

This repository provides the official PyTorch implementation of our paper "Spiral Generative Network for Image Extrapolation".

Our paper can be found in https://link.springer.com/chapter/10.1007/978-3-030-58529-7_41.

Prerequisites

Getting Started

Installation

ImagineGAN

python train.py --path=$configpath$

For example: python train.py --path=./checkpoints/ImagineGAN/celeba/
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.

python train.py --path=$configpath$

For example: python train.py --path=./checkpoints/SliceGAN/celeba/
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}
}