NMS (non maximum suppression)即非极大值抑制用在目标检测中,要理解NMS首先得知道IOU,NMS就是根据IOU阈值来消除检测出得多余得框。
对于目标检测NN会预测出很多的框,其中框与框之间都有很多重复的地方,NMS用来消除冗余的框算法步骤如下:
1) 对所有预测的框按置信度得分降序排列
2) 计算得分最高的框与其他所有框的IOU,如果该IOU大于设定阈值则将对应的框pass,
3) 计算第二得分高的与其他剩余框的IOU,重复第二步,直到遍历完所有的框
NMS (non maximum suppression)即非极大值抑制用在目标检测中,要理解NMS首先得知道IOU,NMS就是根据IOU阈值来消除检测出得多余得框。
对于目标检测NN会预测出很多的框,其中框与框之间都有很多重复的地方,NMS用来消除冗余的框算法步骤如下:
1) 对所有预测的框按置信度得分降序排列
2) 计算得分最高的框与其他所有框的IOU,如果该IOU大于设定阈值则将对应的框pass,
3) 计算第二得分高的与其他剩余框的IOU,重复第二步,直到遍历完所有的框