目标检测--Improving Object Detection With One Line of Code

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/zhangjunhit/article/details/70262784

Improving Object Detection With One Line of Code
https://arxiv.org/abs/1704.04503

Code: https://github.com/bharatsingh430/soft-nms

本文针对目标检测,对非极大值抑制这个环节进行了改善,可以检测出两个相邻的相同物体。
对于目标检测问题,我们目前主流的方法过程如下:
这里写图片描述

如果两个相似物体是相邻的话,以前的非极大值抑制就会只检测出一个目标。
这里写图片描述
原来的非极大值抑制我们在一个区域只取一个最大的矩形框,附近的矩形框置信度全部清零。这里我们提出根据两个矩形框的重合面积来降低非最大矩形框的置信度。这里我们称之为 Soft-NMS

这里写图片描述

这里我们用绿色框的代码取代红色框的代码。

这里写图片描述

这里写图片描述

这里写图片描述

这里写图片描述

展开阅读全文

没有更多推荐了,返回首页