TransT
TransT copied to clipboard
matcher.py里的一些问题
学长好,我想问一下matcher.py里是怎么计算box的,为什么要乘上num_queries开方后的数据
bs, num_queries = outputs["pred_logits"].shape[:2]
for i in range(bs):
cx, cy, w, h = targets[i]['boxes'][0]
cx = cx.item(); cy = cy.item(); w = w.item(); h = h.item()
xmin = cx-w/2; ymin = cy-h/2; xmax = cx+w/2; ymax = cy+h/2
len_feature = int(np.sqrt(num_queries)) # 这句话的作用是什么
Xmin = int(np.ceil(xminlen_feature))
Ymin = int(np.ceil(yminlen_feature))
Xmax = int(np.ceil(xmaxlen_feature))
Ymax = int(np.ceil(ymaxlen_feature))
归一化处理