非极大抑制(Non Maximal Suppression)

最近在看SSD检测网络,其中关于非极大抑制(Non Maximal Suppression)概念

非极大抑制主要是用来解决检测结果中,有多个区域重叠问题,如下图:

非极大抑制对于有相交的box,根据score和box的坐标信息,从中找到置信度比较高的bounding box;对于没相交的就直接保留下来,作为最后结果。

代码实现,参考https://github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/lite/kernels/detection_postprocess.cc中NonMaxSuppressionMultiClass函数。

猜你喜欢

转载自blog.csdn.net/u010349092/article/details/81290850
今日推荐