PySGG
PySGG copied to clipboard
config file for sgdet and sgcls task.
Hi, I am very appreciated with your work.
I confirmed that the author uploaded the predcls-config information at the following issue
https://github.com/SHTUPLUS/PySGG/issues/5
But, I still want to reproduce the result described in the papers for sgdet and sgcls too.
Could you share the config files for those task ?
Thank you a lot.
Hi!
Thanks for interest in our work!
The SGDet config is already in this repo in configs/e2e_relBGNN_vg.yaml. For the SGCls task, you can simply modify the config of PredCls by setting "MODEL.ROI_RELATION_HEAD.USE_GT_OBJECT_LABEL" to False.