跳过正文
  1. Posts/

非极大值抑制(Non-Maximum Suppression,NMS)

NMS是为了在诸多CV任务如边缘检测,目标检测等,找到局部最大值

其主要思想是先设定一个阈值,然后计算检测框的IOU(所谓IOU,也就是intersection-over-union,指的是相交面积除以相并面积,是来衡量overlap程度的指数)。如果IOU大于阈值,说明overlap过大,我们要通过某种算法来将其剔除。

比如下图,在经典的人脸识别任务中,出现了多个检测框,每个检测框有一个置信度confidence,我们通过某个算法,保留一个最好的。

Non-Maximum Suppression(根据正文上下文重绘的示例图,并非遗失原图)

顺便说一下算法的实现步骤把,其实不太重要。就是贪心。

其基本操作流程如下:

  • 首先,计算每一个 bounding box 的面积:

  • (x1, y1) ⇒ 左上点的坐标,(x2, y2) ⇒ 右下点的坐标;

  • (x2-x1+1)x(y2-y1+1)

  • 根据 scores 进行排序(一般从小到大),将 score 最大的bounding box置于队列,接下来计算其余 bounding box 与当前 score 最大的 bounding box 的 IoU,抑制(忽略也即去除)IoU大于设定阈值的 bounding box;

  • 重复以上过程,直至候选 bounding boxes 为空;

最后上一段python代码吧…也很简单,直接转载了别人的…

 1def nms(dets, thresh):
 2    x1 = dets[:, 0]
 3    y1 = dets[:, 1]
 4    x2 = dets[:, 2]
 5    y2 = dets[:, 3]
 6    scores = dets[:, 4]
 7
 8    areas = (x2 - x1 + 1) * (y2 - y1 + 1) # 每个boundingbox的面积
 9    order = scores.argsort()[::-1] # boundingbox的置信度排序
10    keep = [] # 用来保存最后留下来的boundingbox
11    while order.size > 0:     
12        i = order[0] # 置信度最高的boundingbox的index
13        keep.append(i) # 添加本次置信度最高的boundingbox的index
14
15        # 当前bbox和剩下bbox之间的交叉区域
16        # 选择大于x1,y1和小于x2,y2的区域
17        xx1 = np.maximum(x1[i], x1[order[1:]])
18        yy1 = np.maximum(y1[i], y1[order[1:]])
19        xx2 = np.minimum(x2[i], x2[order[1:]])
20        yy2 = np.minimum(y2[i], y2[order[1:]])
21
22        # 当前bbox和其他剩下bbox之间交叉区域的面积
23        w = np.maximum(0.0, xx2 - xx1 + 1)
24        h = np.maximum(0.0, yy2 - yy1 + 1)
25        inter = w * h
26
27        # 交叉区域面积 / (bbox + 某区域面积 - 交叉区域面积)
28        ovr = inter / (areas[i] + areas[order[1:]] - inter)
29        #保留交集小于一定阈值的boundingbox
30        inds = np.where(ovr <= thresh)[0]
31        order = order[inds + 1]
32
33    return keep
34print nms(dets, thresh)

参考链接

目标窗口检测算法-NMS非极大值抑制

相关文章

分类评价指标之Cumulative Match Characteristi (CMC)曲线

CMC曲线全称是Cumulative Match Characteristic (CMC) curve,也就是累积匹配曲线,同ROC曲线Receiver Operating Characteristic (ROC) curve一样,是模式识别系统,如人脸,指纹,虹膜等的重要评价指标,尤其是在生物特征识别系统中,一般同ROC曲线( 多标签图像分类任务的评价方法-mean average precision(mAP) 以及top x的评价方法)一起给出,能够综合评价出算法的好坏。

光流法初探

算是CV领域的传统算法了 只写两句话就够了。 它是空间运动物体在观察成像平面上的像素运动的瞬时速度,是利用图像序列中像素在时间域上的变化以及相邻帧之间的相关性来找到上一帧跟当前帧之间存在的对应关系,从而计算出相邻帧之间物体的运动信息的一种方法。