[发明专利]基于DSP的非极大值抑制算法的实现方法

权利要求书

1.一种基于DSP的非极大值抑制算法的实现方法，其特征在于，包括：

优化NMS算法的步骤，包括如下：

S1：获取K个输入的候选框，并初始化N=1；

S2：判断K和N值的大小，若N≤K，进行到S3，否则，输出候选框集合；

S3：访问第N个候选框B；

S4：找到第N+1至第K个候选框中置信度最高的候选框M；

S5：判断候选框M的置信度P_M是否大于候选框B的置信度P_B，若是，交换候选框M和B在队列中的位置，进行到S6，否则，进行到S6；

S6：判断P_B是否大于置信度阈值P_th，若是，进行到S7，否则将N 更新为N+1，并返回S2；

S7：依次访问第N+1至第K个候选框，计算所访问的候选框和候选框B的交并比IoU；

S8：判断所访问的候选框和候选框B是属于同类别，若是，则进行到S9，否则，将N更新为N+1，并返回S2；

S9：判断所计算的交并比IoU是否大于预设阈值，若是，对所述交并比对应的候选框的置信度附权重，并更新置信度，此后返回S2，否则，将N更新为N+1，并返回S2；

NMS算法移植的步骤，至少包括：对NMS算法所占数据空间的评估、DSP初始化环境设置、及重新构建符合DSP精度标准的NMS算法中候选框数据的存储结构。

2.根据权利要求1所述的基于DSP的非极大值抑制算法的实现方法，其特征在于，所述实现方法还包括实现多种类型的NMS算法的调用函数的步骤，多种类型的NMS算法包括传统NMS算法、线性NMS算法和高斯NMS算法；

在所述S9中的附权重之前，选择对应类型的NMS算法，以获取对应的权重。

3.根据权利要求2所述的基于DSP的非极大值抑制算法的实现方法，其特征在于，在选择传统NMS算法时，获取对应的权重为0；

在选择线性NMS算法时，获取对应的权重为1-IoU；

在选择高斯NMS算法时，获取对应的权重为，其中σ为高斯函数参数。

4.根据权利要求1所述的基于DSP的非极大值抑制算法的实现方法，其特征在于，

对NMS算法所占数据空间的评估，采用默认长度的静态存储方式。

5.根据权利要求1所述的基于DSP的非极大值抑制算法的实现方法，其特征在于，

所述DSP初始化环境设置，包括选取DSP处理器的型号、初始化DSP参数、添加配置文件、分配存储空间大小，并设定所述DSP处理器时钟速率。

6.根据权利要求1所述的基于DSP的非极大值抑制算法的实现方法，其特征在于，所述实现方法还包括：

优化DSP上NMS算法的步骤，至少包括将一个数的除法运算转换为乘以所述数的倒数；

优化DSP编译环境的步骤，至少包括：选择DSP优化选项、定义NMS算法中候选框数据的存储结构为32位浮点型数据的存储结构、以及将所述候选框的数据字节对齐。

7.根据权利要求6所述的基于DSP的非极大值抑制算法的实现方法，其特征在于，在优化DSP上NMS算法的步骤中，所述数为32位浮点型数据F32bit，将F32bit的除法运算转换为乘以F32bit的倒数，具体为：

将区间[0,1)划分为4096等份，建立查找表；

通过查找表查找1/(1+frac)的值，其中frac为F32bit的小数部分数值；

利用求取，其中sign为F32bit的符号位数值、exp表示指数部分数值。

8.根据权利要求6所述的基于DSP的非极大值抑制算法的实现方法，其特征在于，在优化DSP上NMS算法的步骤中，利用DSP处理器内部求倒数函数求所述数的倒数。