Add a download link of trained model to README

YangLing0818 / SGDiff

Official implementation for "Diffusion-Based Scene Graph to Image Generation with Masked Contrastive Pre-Training" https://arxiv.org/abs/2211.11138

51 stars 6 forks source link

Add a download link of trained model to README #14

Closed Maelic closed 3 months ago

Maelic commented 3 months ago

I added a download link of my checkpoint obtained after retraining the model for 335 epochs, performance is 23.54 FID and 18.02 IS. It could be interesting for people who don't have the time or resources to retrain the model from scratch.