/Cross-Modal-Translation

Enhancing Cross-Modal Translation through Cascade Attention Mechanisms

Primary LanguagePythonMIT LicenseMIT

Enhancing Cross-Modal Translation through Cascade Attention Guided Residue GAN

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/.