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

说明书

本发明实施例公开了一种基于图像区域的锚框设置方法、设备和介质，涉及图像检测技术领域。其中，方法包括：统计数据集中在每个图像区域内待检测目标的边界框尺寸，并对每个图像区域内待检测目标的边界框尺寸进行分别聚类得到每个图像区域内的锚框尺寸；获取待检测图像经过特征提取得到的特征图；将所述特征图划分为对应不同图像区域的多个子特征图，将多个子特征图分别经过卷积处理得到每个图像区域的直接预测结果，根据每个图像区域的所述锚框尺寸和尺寸缩放比得到每个图像区域的预测框的尺寸。本实施例针对不同的图像区域设置不同尺寸的锚框，增强边缘识别效果，减少无效锚框的数量。

技术领域

本发明实施例涉及图像检测技术，尤其涉及一种基于图像区域的锚框设置方法、设备和存储介质。

背景技术

在目标检测任务中，识别图像中不同区域、不同大小的目标，是在目标检测中通常遇到的一个难题。对此，通常使用锚框的方法来解决。具体而言，就是根据待检测目标的实际尺寸，通过人工选择或聚类的方法得到若干个典型的矩形框，并将这些矩形框设置在图像中若干个固定的点；在检测过程中，目标检测模型对这些设定的矩形框进行微量修正，即可得到更精确的检测结果。

在实际应用场景中，存在很多待检测目标出现在图像边缘的情况，如果仍然使用现有的锚框设置方法，会使得大量的锚框超出了图像的边界，无效锚框数量增多；而且，对图像边缘的目标识别效果相对较差。

发明内容

本发明实施例提供一种基于图像区域的锚框设置方法、设备和存储介质，以针对不同的图像区域设置不同尺寸的锚框，增强边缘识别效果，减少无效锚框的数量。

第一方面，本发明实施例提供了一种基于图像区域的锚框设置方法，包括：

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

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

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

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

第二方面，本发明实施例还提供了一种电子设备，所述电子设备包括：

一个或多个处理器；

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

当所述一个或多个程序被所述一个或多个处理器执行，使得所述一个或多个处理器实现任一实施例所述的基于图像区域的锚框设置方法。

第三方面，本发明实施例还提供了一种计算机可读存储介质，其上存储有计算机程序，该程序被处理器执行时实现任一实施例所述的基于图像区域的锚框设置方法。

本发明实施例考虑到待检测目标出现在不同图像区域时，会由于显示不完整呈现不同的形状特点，从而聚类得到不同图像区域内的锚框尺寸，正是由于锚框的尺寸符合区域内待检测目标的形状特点，很少有锚框会超出图像。通过将对应不同图像区域的子特征图进行卷积处理以及采用对应区域的锚框尺寸进行再处理，得到符合不同区域待检测目标呈现特点的预测框的尺寸。正是由于预测框的尺寸符合区域内待检测目标的形状特点，可以准确地预测位于各图像区域内的目标位置。

附图说明

为了更清楚地说明本发明具体实施方式或现有技术中的技术方案，下面将对具体实施方式或现有技术描述中所需要使用的附图作简单地介绍，显而易见地，下面描述中的附图是本发明的一些实施方式，对于本领域普通技术人员来讲，在不付出创造性劳动的前提下，还可以根据这些附图获得其他的附图。