search

KaihuaTang / Scene-Graph-Benchmark.pytorch

A new codebase for popular Scene Graph Generation methods (2020). Visualization & Scene Graph Extraction on custom images/datasets are provided. It's also a PyTorch implementation of paper “Unbiased Scene Graph Generation from Biased Training CVPR 2020”
MIT License
1.03k stars 228 forks source link

Training with class label subset #85

Open ChrisAlthaus opened 3 years ago

ChrisAlthaus commented 3 years ago

❓ Questions and Help

Hi, I wonder if it's possible to train the SGDet Model with custom class labels (subset of the original 150 labels) only to improve model performance just on this set of class labels? So far i can image to implement this by 2 methods:

  1. Filter all box & relation annotations of the scene graphs in vg/VG-SGG-with-attri.h5
  2. Modify load_graphs in /Scene-Graph-Benchmark.pytorch/maskrcnn_benchmark/data/datasets/visual_genome.py to filter out all unneeded class labels. (This approach i am implemented at the moment, but I am getting IndexError in get_groundtruth)

Thanks for this well documented repository and I'm looking forward for any suggestions.

pcyyo commented 3 years ago

have you solved this problem？