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.06k stars 229 forks source link

about vg dataset's dataloader bug #171

Open JackWhite-rwx opened 2 years ago

JackWhite-rwx commented 2 years ago

🐛 Bug

First of all, thank you very much for your excellent work! I finds a bug at

https://github.com/KaihuaTang/Scene-Graph-Benchmark.pytorch/blob/45cd54f7465b81d3154e94fcab2b554a09637f6f/maskrcnn_benchmark/data/datasets/visual_genome.py#L87

Although it currently does not affect the normal operation of the code,I think that can be fixed:

target = self.get_groundtruth(index, flip_img=flip_img)

to solve some problem about Parameter Passing in the future.

thanks again.