search

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

VG Links broken, Hardware information #13

Open marquies opened 4 months ago

marquies commented 4 months ago

Hi, thank you for your work. While testing your model, I recognized some VG links in scripts/download_vg.sh are not working. These are okay for me:

wget https://homes.cs.washington.edu/~ranjay/visualgenome/data/dataset/objects.json.zip -O $VG_DIR/objects.json.zip
wget https://homes.cs.washington.edu/~ranjay/visualgenome/data/dataset/attributes.json.zip -O $VG_DIR/attributes.json.zip
wget https://homes.cs.washington.edu/~ranjay/visualgenome/data/dataset/relationships.json.zip -O $VG_DIR/relationships.json.zip
wget https://homes.cs.washington.edu/~ranjay/visualgenome/data/dataset/object_alias.txt -O $VG_DIR/object_alias.txt
wget https://homes.cs.washington.edu/~ranjay/visualgenome/data/dataset/relationship_alias.txt -O $VG_DIR/relationship_alias.txt
wget https://homes.cs.washington.edu/~ranjay/visualgenome/data/dataset/image_data.json.zip -O $VG_DIR/image_data.json.zip

I tried with GPUs of 3GB, 16GB and 24GB memory. Only 24 GB could run the training. Maybe this information is worth to mention in the readme.