How can I get gt_masks in test stage of the Predicate Classification task？

Jingkang50 / OpenPSG

Benchmarking Panoptic Scene Graph Generation (PSG), ECCV'22

https://psgdataset.org

MIT License

407 stars 68 forks source link

How can I get gt_masks in test stage of the Predicate Classification task？ #96

Open wyz-gitt opened 1 year ago

wyz-gitt commented 1 year ago

How do I modify the test_pipeline in this configuration file (configs/base/datasets/psg.py) during the testing phase to get the value of the image gt_masks in (open/models/frameworks/sg_panopticfpn.py, in line 340 def detector_simple_test).

zgerem commented 1 year ago

hey, could you find a solution to that?