[发明专利]一种自适应选择交并区域比阈值的非极大值抑制方法

权利要求书

1.一种自适应选择交并区域比阈值的非极大值抑制方法，其特征在于，包括：

初始化设定多个IOU阈值；

过滤置信度小于指定阈值的预测边框，并根据目标类型分为不同类的集合；

对所有类的集合以及集合中的所有元素执行如下操作：选择一个类的集合，取出置信度最高的预测边框，依次计算与集合中其他预测边框的IOU，综合根据两个预测边框的中心点绝对距离、所属特征层下采样率与训练正样本选择策略，动态设置IOU阈值，并从集合中删除超过阈值的预测边框，执行预测边框非极大值抑制操作；完成所有类别的非极大值抑制操作后输出目标检测结果。

2.如权利要求1所述的一种自适应选择交并区域比阈值的非极大值抑制方法，其特征在于，根据目标类型分为不同类的集合，具体为：根据预测边框的类别将所有保留即参与NMS的预测边框分为属于不同类的集合。

3.如权利要求2所述的一种自适应选择交并区域比阈值的非极大值抑制方法，其特征在于，在集合内根据预测边框的置信度进行降序排序。

4.如权利要求1所述的一种自适应选择交并区域比阈值的非极大值抑制方法，其特征在于，初始化设定3个IOU阈值，分别为min，0.5和max，其中min的取值不应小于0.3，max的取值不应大于0.6，0.5则对应标准NMS方法的缺省阈值。

5.如权利要求4所述的一种自适应选择交并区域比阈值的非极大值抑制方法，其特征在于，设定集合中置信度最高的预测边框B0所在特征层的下采样率为集合中其他预测边框Bn所在特征层的下采样率为且目标检测网络的正样本选择策略为选择位于真实标注边框内、且距离真实标注边框中心小于的采样点作为正样本的中心点；如果预测边框B0与预测边框Bn的中心距离小于且小于则IOU阈值设为max即最大的IOU阈值。

6.如权利要求5所述的一种自适应选择交并区域比阈值的非极大值抑制方法，其特征在于，如果预测边框B0与预测边框Bn的中心距离D大于且大于同时|S₀-S_n|大于1，则IOU阈值设为min即最小的IOU阈值。

7.如权利要求5所述的一种自适应选择交并区域比阈值的非极大值抑制方法，其特征在于，如果预测边框B0与预测边框Bn的中心距离D不小于或不小于以及预测边框B0与预测边框Bn的中心距离D不大于或不大于或|S₀-S_n|不大于1，则IOU阈值设为0.5。

8.一种计算机可读存储介质，其特征在于，所述计算机存储介质中包含一个或多个程序指令，所述一个或多个程序指令用于被处理器执行如权利要求1-7任一项所述的一种自适应选择交并区域比阈值的非极大值抑制方法。