search

JindongJiang / latent-slot-diffusion

Official Release of NeurIPS 2023 Spotlight paper "Object-Centric Slot Diffusion"
https://latentslotdiffusion.github.io/
MIT License
59 stars 11 forks source link

Instructions for COCO dataset #1

Open kaanakan opened 7 months ago

kaanakan commented 7 months ago

Hello,

thank you for sharing your code. It's an amazing paper.

I would like to ask if you can share any instructions for COCO dataset clarifying how to download and process the dataset.

Thanks in advance. Kind regards.

JindongJiang commented 7 months ago

Hello @kaanakan,

I recall that for the COCO dataset, we only need to download the 2017 training image from http://images.cocodataset.org/zips/train2017.zip. After unzipping the file, provide the unzipped path as --dataset_root as in configs/coco/train.sh.

kaanakan commented 7 months ago

Thank you for the fast response. Can I also ask how you handled the validation part? Specifically, how can I reproduce the results on table 4 of appendix (Quantitative Evaluation of Stable-LSD on COCO dataset)? Thanks in advance.