search

Jingkang50 / OpenPSG

Benchmarking Panoptic Scene Graph Generation (PSG), ECCV'22
https://psgdataset.org
MIT License
407 stars 68 forks source link

Support of VG-150 Dataset #81

Open ShunchiZhang opened 1 year ago

ShunchiZhang commented 1 year ago

Hi, I noticed that this codebase doesn't support VG-150 Dataset.

https://github.com/Jingkang50/OpenPSG/blob/ae418a15f6c0d558b665ffaca3ec0dd52e0f6f24/README.md?plain=1#L248-L251

Do you have any plans to support it in the future? I'm glad to contribute if you'd like to sync the progress of this feature with me.

Jingkang50 commented 1 year ago

Yes, @king159 and @wzy27 are trying to support VG-150. I will discuss with them when they finish their final exams.

ShunchiZhang commented 1 year ago

Yes, @king159 and @wzy27 are trying to support VG-150.

I will discuss with them when they finish their final exams.

That would be nice. Thanks!

Yassin-fan commented 1 year ago

Still want to ask the same question, it would be wonderful to seethe progress in VG-150.

Jingkang50 commented 1 year ago

Please download the VG dataset here, and put it into ./data dir. We have https://github.com/Jingkang50/OpenPSG/blob/main/openpsg/datasets/sg.py to process the dataset.

Yassin-fan commented 1 year ago

Thanks to your work! Does it mean that we can now train detr-like models proposed in the paper on the VG-150 dataset? I have put VG-100K in ./data and here is my dir, is this right? ├── data │ ├──vg150.json │ ├── VG_100K │ │ ├── XX.jpg Then what should I do to train a detr-like model on VG-150? I didn't found the settings of dataset kinds and dirs in config files. Sorry to bother you and I'm not familiar with this framework.