Semantic Diversity-aware Prototype-based Learning for Unbiased Scene Graph Generation (ECCV 2024)

This is the official code for ECCV 2024 paper Semantic Diversity-aware Prototype-based Learning for Unbiased Scene Graph Generation.

Please wait a bit longer for the repository to be completed.

Installation

Check INSTALL.md for installation instructions.

Dataset

Check DATASET.md for instructions of dataset preprocessing.

Pretrained Models

For VG dataset, the pretrained object detector we used is provided by Scene-Graph-Benchmark, you can download it from this link. For GQA dataset, we trained a new object detector which can be downloaded from this link.

Put the checkpoint into the folder:

mkdir -p checkpoints/detection/pretrained_faster_rcnn/
mv /path/gqa_det.pth checkpoints/detection/pretrained_faster_rcnn/
mv /path/gqa_det.pth checkpoints/detection/pretrained_faster_rcnn/

Please wait a moment until the repository is complete.