R-CNN算法中NMS的具体做法

假设有20类,2000个建议框,最后输出向量维数2000*20,则每列对应一类,一行是各个建议框的得分,NMS算法步骤如下:
 ① 对2000×20维矩阵中每列按从大到小进行排序;
 ② 从每列最大的得分建议框开始,分别与该列后面的得分建议框进行IoU计算,若IoU>阈值,则剔除得分较小的建议框,否则认为图像中存在多个同一类物体;
 ③ 从每列次大的得分建议框开始,重复步骤②;
 ④ 重复步骤③直到遍历完该列所有建议框;
 ⑤ 遍历完2000×20维矩阵所有列,即所有物体种类都做一遍非极大值抑制;
 ⑥ 最后剔除各个类别中剩余建议框得分少于该类别阈值的建议框。

原文地址:https://www.cnblogs.com/zhaopengpeng/p/12268193.html