Yumeng/ALDM

Adversarial Supervision Makes Layout-to-Image Diffusion Models Thrive (ICLR 2024)

Project Page | ArXiv | Code

This model repo contains checkpoints trained on Cityscapes and ADE20K datasets using methods proposed in ALDM. For usage instructions, please refer to our Github.

Model information

ade20k_step9.ckpt and cityscapes_step9.ckpt are pretrained diffusion model weights for inference.

encoder_epoch_50.pth, decoder_epoch_50_20cls.pth and decoder_epoch_50_151cls.pth are segmentation models used for discriminator intialization in training, which are adopted from pretrained UperNet101 here.