最近在看SSD检测网络,其中关于非极大抑制(Non Maximal Suppression)概念
非极大抑制主要是用来解决检测结果中,有多个区域重叠问题,如下图:
非极大抑制对于有相交的box,根据score和box的坐标信息,从中找到置信度比较高的bounding box;对于没相交的就直接保留下来,作为最后结果。
代码实现,参考https://github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/lite/kernels/detection_postprocess.cc中NonMaxSuppressionMultiClass函数。