目标检测中的LOU(交并比)和NMS(非极大值抑制)代码实现
代码如下所示:#假设box1的维度为[N,4] box2的维度为[M,4]def Lou(box1, box2): N = box1.size(0) M = box2.size(0) #计算两个边框的交集,首先获取相交的左上和右下的坐标 lt = torch.max( box1[:,:2].unsqueeze(1).expend(N, M, 2), box2[:.
原创
2022-11-01 19:19:07 ·
851 阅读 ·
0 评论