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

Howto get 'image_graph' and 'text_graph' #104

Open ihaeyong opened 3 years ago

ihaeyong commented 3 years ago

https://github.com/KaihuaTang/Scene-Graph-Benchmark.pytorch/blob/0482daef90c98f810774aab042005add6859f529/maskrcnn_benchmark/image_retrieval/dataloader.py#L72

'image_graph' and 'text_graph' seems to be related to preprocessing in image retrieval task. Could you provide the full preprocessing file?

KaihuaTang commented 3 years ago

I think https://github.com/KaihuaTang/Scene-Graph-Benchmark.pytorch/issues/64 may partially answer this question.