princeton-vl / px2graph

Training code for "Pixels to Graphs by Associative Embedding"
BSD 3-Clause "New" or "Revised" License
133 stars 22 forks source link

score_loss for relations #8

Open beihangwenbin opened 6 years ago

beihangwenbin commented 6 years ago

I find your code for counting rel score_loss(graph.py/539) is: score_loss = calc.bce_loss(tf.squeeze(relscores[:opt.obj_slotsnum_objs],1), rel_scores_gt[:opt.obj_slotsnum_objs]) I think opt.obj_slots and num_objs should be replaced by opt.rel_slots and num_rels.