jlim262 / toy-cyclegan

CycleGAN (image to image translation) implementation in pytorch

toy-cyclegan

Simplified PyTorch implementation of CycleGAN based on the official implementaton.

Usage

Clone this repository

$ git clone http://github.com/jlim262/toy-cyclegan.git
$ cd toy-cyclegan

Prepare your dataset

In order to translate Bart Simpson to Lisa Simpson, Download Bart Simpson images into ./dataset/bart2lisa/trainA and Lisa Simpson into ./dataset/bart2lisa/trainB

Train a model

$ python train.py

View results on tensorboard

$ tensorboard --logdir=runs