Skip to content

colemiller94/gatedgan

Repository files navigation

Gated-Gan for Multi-Style Transfer

PyTorch Implementation of https://github.com/xinyuanc91/Gated-GAN
The goal of this project is to translate (from lua to python) and train the model described in this paper: https://arxiv.org/abs/1904.02296
I use the general architecture of this repository as a baseline: https://github.com/aitorzip/PyTorch-CycleGAN
and modify it to the paper's specifications.

Notebooks

  • Builder Book: Train the model
  • Demo: Interact with most recent model using jpeg/jpg in working directory

Requirements

Only tested on Python 3.7

Follow the instructions in pytorch.org for your current setup

Visdom (training only)