attribute-guided-image-generation-from-layout

This is the code repository complementing the paper attribute-guided image generation from layout. The pretrained model can be downloaded here: (TODO)

Dependencies and Setup

cd data/Dataset/vg
chmod +x download_vg.sh
./download_vg.sh

Preprocessing:

cd data/Dataset/vg
python train_test_split.py # split the dataset
cd ../..
python preprocess_vg.py

To train 64x64 (or 128x128) from scratch, run

python train64.py

The script will check if pretrained models exist.

Training 128x128 model is computationally expensive. You might consider using torch.DataParallel for training with a larger batch size.

To test 64x64 (or 128x128) model, run

python test64.py

You can modify attributes as you wish in the test64.py (test128.py) script.