[发明专利]基于图像区域的锚框设置方法、设备和存储介质

权利要求书

1.一种基于图像区域的锚框设置方法，其特征在于，包括：

统计数据集中在每个图像区域内待检测目标的边界框尺寸，并对每个图像区域内待检测目标的边界框尺寸进行分别聚类得到每个图像区域内的锚框尺寸；所述数据集包括多张图像；

获取待检测图像经过特征提取得到的特征图；所述待检测图像包括多个不重叠的图像区域；

将所述特征图划分为对应不同图像区域的多个子特征图，将所述多个子特征图分别经过卷积处理得到每个图像区域的直接预测结果，所述直接预测结果至少包括在每个图像细胞单元内偏移的预测框相对于锚框的尺寸缩放比；

根据每个图像区域的所述锚框尺寸和尺寸缩放比得到每个图像区域的预测框的尺寸，所述预测框的尺寸用于预测所述待检测目标的真实尺寸。

2.根据权利要求1所述的方法，其特征在于，所述统计数据集中在每个图像区域内待检测目标的边界框尺寸，包括：

获取所述多张图像中待检测目标的边界框中心位置，并基于所述待检测目标的边界框中心位置汇总每个图像区域内出现的待检测目标；

统计每个图像区域内出现的待检测目标的边界框尺寸。

3.根据权利要求1所述的方法，其特征在于，所述多个图像区域包括边缘区域和中间区域；

所述基于所述待检测目标的中心位置汇总每个图像区域内出现的待检测目标，包括：

将中心位置出现在边缘区域的待检测目标作为边缘区域内出现的待检测目标；

将中心位置出现在图像中的待检测目标作为中间区域内出现的待检测目标。

4.根据权利要求3所述的方法，其特征在于，所述边缘区域包括左右边缘区域、上下边缘区域和角点区域；

所述将中心位置出现在边缘区域的待检测目标作为边缘区域内出现的待检测目标，包括：

将中心位置出现在左右边缘区域的待检测目标作为左右边缘区域内出现的待检测目标；

将中心位置出现在上下边缘区域的待检测目标作为上下边缘区域内出现的待检测目标；

将中心位置出现在角点区域的待检测目标作为角点区域内出现的待检测目标。

5.根据权利要求1所述的方法，其特征在于，所述将所述多个子特征图分别经过卷积处理得到每个图像区域的直接预测结果，包括：

根据每个子特征图对应图像区域内锚框尺寸的个数，确定卷积核的个数；

将多个子特征图分别经过对应数量的卷积核得到直接预测结果，所述直接预测结果至少包括在每个图像细胞单元内偏移的预测框相对于锚框的尺寸缩放比；每个图像细胞单元内偏移的预测框个数与所述锚框尺寸的个数一致。

6.根据权利要求5所述的方法，其特征在于，同一图像区域内，所述每个图像细胞单元内偏移的预测框与锚框尺寸具有一一对应关系；

所述根据每个图像区域的所述锚框尺寸和尺寸缩放比得到每个图像区域的预测框的尺寸，包括：

在同一图像区域中，将每个图像细胞单元内偏移的每个预测框相对于锚框的尺寸缩放比乘以对应的锚框尺寸，得到每个图像区域的预测框的尺寸。

7.根据权利要求1所述的方法，其特征在于，在所述根据每个图像区域的所述锚框尺寸和尺寸缩放比得到每个图像区域的预测框的尺寸之后，还包括：

按照图像区域的位置，将多个图像区域的预测框的尺寸进行合并；

所述合并后的多个预测框的尺寸用于预测所述待检测目标的真实尺寸。

8.一种电子设备，其特征在于，包括：

一个或多个处理器；

存储器，用于存储一个或多个程序，

当所述一个或多个程序被所述一个或多个处理器执行，使得所述一个或多个处理器实现如权利要求1-7中任一所述的基于图像区域的锚框设置方法。

9.一种计算机可读存储介质，其上存储有计算机程序，其特征在于，该程序被处理器执行时实现如权利要求1-7中任一所述的基于图像区域的锚框设置方法。