PyTorch实现非极大值抑制(NMS)

NMS即non maximum suppression即非极大抑制，顾名思义就是抑制不是极大值的元素，搜索局部的极大值。在最近几年常见的物体检测算法（包括rcnn、sppnet、fast-rcnn、faster-rcnn等）中，最终都会从一张图片中找出很多个可能是物体的矩形框，然后为每个矩形框为做类别分类概率。本文来通过Pytorch实现NMS算法。

如果你在做计算机视觉(特别是目标检测)，你肯定会听说过非极大值抑制(nms)。网上有很多不错的文章给出了适当的概述。简而言之，非最大抑制使用一些启发式方法减少了输出边界框的数量，例如交叉除以并集(iou)。