实例分割：IOU与NMS笔记

阅读量：

文章结构概览

1. IOU
- - 1.1 IOU
  - 1.1 GIOU
- 2. nms
- - 2.1 vanilla nms
  - 2.2. soft NMS
  - 2.3 fast nms
  - 2.4 matrix nms

IOU指标分析与应用

1.1 `IOU`

参考：知乎
由于难以直接把握多个维度下多个边界框之间的交并比，因此决定逐步进行推导。
pytoch 源代码

复制代码

    =
    
    # IOU计算
    # 假设box1维度为[N,4]   box2维度为[M,4]
    def iou(self, box1, box2):
        N = box1.size(0)
        M = box2.size(0)
    
        lt = torch.max(  # 左上角的点 注意：是inter的左上角的点，因此才要算MAX
            box1[:, :2].unsqueeze(1).expand(N, M, 2),   # [N,2]->[N,1,2]->[N,M,2]

全部评论 (0)

还没有任何评论哟~