Skip to content
/ lnfmm Public

Latent Normalizing Flows for Many-to-Many Cross Domain Mappings (ICLR 2020)

License

Notifications You must be signed in to change notification settings

visinf/lnfmm

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Introduction

This repository is the PyTorch implementation of the paper:

Latent Normalizing Flows for Many-to-Many Cross Domain Mappings (ICLR 2020)

Shweta Mahajan, Iryna Gurevych, Stefan Roth

Getting started

This code has been developed under Python 3.5, Pytorch 1.0.0 and CUDA 10.0.

  1. Please run requirements.py to check if all required packages are installed.
  2. The dataset used in this project is COCO 2014. The dataset is available here.

Training

The script train.py is used for training. The parameters are listed in params.json. Note that there are two different configurations for best performance on the image captioning and text-to-image synthesis tasks.

Example usage to train a model on COCO 2014 for captioning is,

python train.py --config params_i2t

Example usage to train a model on COCO 2014 for text-to-image synthesis task is,

python train.py --config params_t2i

Note that for training CUDA 10.0 and GPU devices are required. The number of GPUs used can be set in params.json. Also note that we use 1 Nvidia Volta V100 GPU and 3 Nvidia Volta V100 GPUs with 32GB for the captioning and text-to-image synthetis tasks respectively.

Generation and Validation

For evalutaion we use the following repos,

  1. Oracle - We use the version of pycocoeval cap which supports Python 3 available here.
  2. Concensus Reranking - We use the repo of mRNN-CR.
  3. Diversity - We use the repo of DiversityMetrics (requires Python 2.7).

Checkpoints are available for text-to-image synthesis and for image captioning.

Bibtex

@inproceedings{mahajan2020latent,
title = {Latent Normalizing Flows for Many-to-Many Cross-Domain Mappings},
author = {Mahajan, Shweta and Gurevych, Iryna and Roth, Stefan},
booktitle = {International Conference on Learning Representations},
year = {2020},
}