一种自适应选择交并区域比阈值的非极大值抑制方法

摘要

本发明公开一种自适应选择交并区域比阈值的非极大值抑制方法。所述方法包括：初始化设定多个IOU阈值；过滤置信度小于指定阈值的预测边框，并根据目标类型分为不同类的集合；对所有类的集合以及集合中的所有元素执行如下操作：选择一个类的集合，取出置信度最高的预测边框，依次计算与集合中其他预测边框的IOU，综合根据两个预测边框的中心点绝对距离、所属特征层下采样率与训练正样本选择策略，动态设置IOU阈值，并从集合中删除超过阈值的预测边框，执行预测边框非极大值抑制操作；完成所有类别的非极大值抑制操作后输出目标检测结果。采用本申请技术方案，能够过滤重复的预测边框，降低误检率，且能够避免小目标的漏检，提高小目标检测概率。