KaiChen1998 / GeoDiffusion

Official PyTorch implementation of GeoDiffusion in ICLR 2024 (https://arxiv.org/abs/2306.04607)
https://kaichen1998.github.io/projects/geodiffusion/
MIT License
61 stars 3 forks source link

COCO dataset's Trainablity? #19

Closed why529913 closed 3 months ago

why529913 commented 3 months ago

I see that nuimages' Trainablity is trained with 800x456 GeoDiffusion. Is the COCO dataset's Trainablity also trained with 800x456 GeoDiffusion input size? Or is it 512x512?

KaiChen1998 commented 3 months ago

The basic idea is to keep the training resolution consistent with the resolution of the generated images.

Therefore yes, the training size on COCO experiments is also 800x456.