Closed taotaotao0412 closed 1 year ago
You can modify one line at here to add your labels. Note that your provided labels should be aligned with VG's object class ids.
Then modify the MODEL.ROI_RELATION_HEAD.USE_GT_OBJECT_LABEL
to True
in the VG1800-SGCLS or VG50-SGCLS.
OK, Thanks a lot
Nice job! I want to get more precious detection result because Faster RCNN with low accurate in some complex scene. I want to get scene graph via image, bbox and label. How should I do. THANKS FOR YOUR HELP!