jwyang / graph-rcnn.pytorch

[ECCV 2018] Official code for "Graph R-CNN for Scene Graph Generation"
732 stars 157 forks source link

Evaluation method and configs #100

Open xuewyang opened 4 years ago

xuewyang commented 4 years ago

Hi, When I evaluate using this chk: sg_baseline_joint_ckpt.pth. The result is not good. I use this config file: sgg_res101_joint.yaml, the parameters are set as: MODEL: WEIGHT_IMG: "catalog://ImageNetPretrained/MSRA/R-101" WEIGHT_DET: "sg_baseline_joint_ckpt.pth" RELATION_ON: True ALGORITHM: "sg_baseline" USE_FREQ_PRIOR: False USE_RELPN: True

Can you help me to get the correct result? Thank you.