Home

Awesome

GR-GAN

Pytorch implementation for reproducing GR-GAN results in the paper GRADUAL REFINEMENT TEXT-TO-IMAGE GENERATION

<img src="GAN_Model_V4.png" width="900px" height="550px"/>

Usage

Get the code from github:git clone https://github.com/BoO-18/GR-GAN.git

Create a new conda env:conda create -n grgan python=3.7 and conda activate grgan

Install torch==1.7.1: conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0

Folder CLIP is code from OPENAI with some changes to the output of the image encoder and text encoder. You should run:pip install CLIPto install it.

Data

  1. Download our preprocessed metadata for coco and save them to data/
  2. Download coco dataset and extract the images to data/coco/

Training

Pretrained Model

Demo

Validation