/CAR-GAN

code for the paper "Cascade Attention Guided Residue GAN for Cross-Modal Translation"

Primary LanguagePython

Cascade Attention Guided Residue GAN for Cross-Modal Translation

Requirements

  • pytorch 0.4+
  • python 3.6.8
  • visdom 0.1.8.8
  • pillow 5.4.1

Model

Download the model via google drive and put models into checkpoints dir.

Train

$ sh ltrain.sh

Test

$ sh ltest.sh

Acknowledgment

Our code borrows heavily from the the pix2pix implementation https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix/.