[发明专利]基于ATSS的图像处理方法及其装置、电子设备、存储介质

说明书

本发明提供了一种基于ATSS的图像处理方法及其装置、电子设备、存储介质，包括：获取特征图，其中，特征图中预先标注有原标记框；获取特征图的下采样尺度，根据下采样尺度在特征图中生成多个候选框；确定候选框与原标记框之间的第一距离，当第一距离满足预先设定的第一条件，确定候选框为有效候选框；确定有效候选框与原标记框之间的交并比，当交并比满足预先设定的第二条件，确定有效候选框为正样本；根据正样本生成token，并根据token进行自注意权重的计算。根据本实施例的技术方案，能够通过设置好的第一条件和第二条件对候选框进行筛选，舍弃不符合条件的候选框，通过token的裁剪有效减少了token的计算量，提高了图像处理的效率。

技术领域

本发明涉及但不限于人工智能领域，尤其涉及一种基于ATSS的图像处理方法及其装置、电子设备、存储介质。

背景技术

随着人工智能技术的发展，应用于图像识别的模型越来越多，而Transformer模型具有强大的性能，应用越来越广泛。虽然Transformer模型的准确率非常高，但是其核心为自注意机制，在输入图像之后，需要以整个特征图作为token计算出自注意权重，再根据自注意权重与输入图像融合，而Transformer模型的时间复杂度非常高，这会导致计算量较大，影响图像识别的效率。

发明内容

以下是对本文详细描述的主题的概述。本概述并非是为了限制权利要求的保护范围。

本发明实施例提供了一种基于ATSS的图像处理方法及其装置、电子设备、存储介质，能够减少token的计算量，从而提高Transformer模型的效率。

第一方面，本发明实施例提供了一种基于ATSS的图像处理方法，包括：

获取特征图，其中，所述特征图中预先标注有原标记框；

获取所述特征图的下采样尺度，根据所述下采样尺度在所述特征图中生成多个候选框；

确定所述候选框与所述原标记框之间的第一距离，当所述第一距离满足预先设定的第一条件，确定所述候选框为有效候选框；

确定所述有效候选框与所述原标记框之间的交并比，当所述交并比满足预先设定的第二条件，确定所述有效候选框为正样本；

根据所述正样本生成token，并根据所述token进行自注意权重的计算。

在一些实施例中，所述确定所述候选框与所述原标记框之间的第一距离，包括：

获取第一坐标和第二坐标，所述第一坐标为所述候选框的中心点坐标，所述第二坐标为所述原标记框的中心点坐标；

将所述第一坐标和所述第二坐标之间的距离确定为所述第一距离。

在一些实施例中，所述确定所述有效候选框与所述原标记框之间的交并比，包括：

确定第一区域，所述第一区域为所述有效候选框与所述原标记框的交集区域；

确定第二区域，所述第二区域为所述有效候选框和所述原标记框的并集区域；

将所述第二区域的面积与所述第一区域的面积的比例确定为所述交并比。

在一些实施例中，所述获取所述特征图的下采样尺度，包括：

确定所述特征图的当前的采样深度为目标采样深度；

获取预先设定的尺度映射表，所述尺度映射表记载有采样深度与所述下采样尺度的映射关系；

根据所述目标采样深度从所述尺度映射表中确定所述下采样尺度。

在一些实施例中，所述根据所述下采样尺度在所述特征图中生成多个候选框，包括：

根据所述下采样尺度得到采样宽度和采样长度