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

About VG data preprocess #9

Closed HuilingSun closed 7 months ago

HuilingSun commented 7 months ago

When I run preprocess.py, I get the vocab.json file, which looks like this. It shows that the number of objects is 260, not 179 as stated in the paper, and the number of relations is 66, not 46. I think it may be because object_alias.txt and relationships_alias.txt are downloaded from other links, resulting in the wrong number of objects and relationships. We would appreciate it if you could provide a download of these two files, or provide the pre-processed.h5 files directly. 屏幕截图 2023-11-29 194750

HuilingSun commented 7 months ago

问题已解决，下载的object.json文件和relationships.json文件应该是VG dataset v1.4版本。script/download_vg中的下载链接不能用，可以参考此处下载链接：https://homes.cs.washington.edu/~ranjay/visualgenome/api.html