search

NVIDIA / ContrastiveLosses4VRD

Implementation for the CVPR2019 paper "Graphical Contrastive Losses for Scene Graph Generation"
Other
200 stars 41 forks source link

maybe bug in model_builder_rel.py(pytorch1_0 version) ? #22

Open yrcong opened 4 years ago

yrcong commented 4 years ago

Hi,

for the code of pytorch1.0 version, in model_builder_rel.py line 504

_, prd_bias_scores_sbj_pos, _, ttl_cls_scores_sbj_pos, _, _, prd_cls_feats = \ self.RelDN(spo_feat_sbj_pos, spt_feat_sbj_pos, sbj_labels_sbj_pos_fg, obj_labels_sbj_pos_fg, sbj_feat_sbj_pos, obj_feat_sbj_pos)

7 elements are expected but RelDN (reldn_heads.py) only returns 6 (line 136). return prd_vis_scores, prd_bias_scores, prd_spt_scores, ttl_cls_scores, sbj_cls_scores, obj_cls_scores

For the master branch there is not this problem..

Is this only a bug or something that i didn't understand or made mistakes?

thanks a lot