↓ 5 callersMethod__init__(self, hidden_tensor, order_tensor, order_count, dists_tensor,
commitments_tensor, dropout_ma
openpsg/models/relation_heads/approaches/treelstm_util.py:349
↓ 3 callersFunctionencode_box_infoencode proposed box information (x1, y1, x2, y2) to (cx/wid, cy/hei, w/wid, h/hei, x1/wid, y1/hei, x2/wid, y2/hei, wh/wid*hei)
openpsg/models/relation_heads/approaches/motif_util.py:138
↓ 3 callersFunctionintersect_2dGiven two arrays [m1, n], [m2,n], returns a [m1, m2] array where each entry is True if those rows match. :param x1: [m1, n] numpy array :
openpsg/evaluation/sgg_eval_util.py:12
↓ 3 callersMethodrelation_ranking_forward(self, input, det_result, gt_result, num_rels,
is_testing)
openpsg/models/relation_heads/relation_head.py:235
↓ 2 callersMethod__init__(self, config, obj_classes, embed_dim, inputs_dim, hidden_dim,
rnn_drop)
openpsg/models/relation_heads/approaches/motif.py:72
↓ 2 callersFunctionget_box_info input: [batch_size, (x1,y1,x2,y2)] size: [h, w] output: [batch_size, (x1,y1,x2,y2,cx,cy,w,h)]
openpsg/models/relation_heads/approaches/relation_util.py:438
↓ 2 callersFunctionresize_bbox(old_height: int, old_width: int,
bbox: Tuple[int, int, int, int], resize: int)
openpsg/utils/vis_tools/preprocess.py:133
↓ 2 callersFunctionsgg_evaluation(
mode,
groundtruths,
predictions,
iou_thrs,
logger,
ind_to_predicates,
multiple_p
openpsg/evaluation/sgg_eval.py:23