Open Sevenlepus opened 3 years ago
# 计算真实框的位置 gt_box = torch.FloatTensor(torch.cat([torch.zeros_like(gws), torch.zeros_like(ghs), gws, ghs], 1)) # 计算出所有先验框的位置 anchor_shapes = torch.FloatTensor(torch.cat((torch.zeros((self.num_anchors, 2)), torch.FloatTensor(anchors)), 1)) # 计算重合程度 anch_ious = jaccard(gt_box, anchor_shapes)
啊?因为要求真实框和先验框的重合程度,然后筛选先验框中最重合的一个框