[发明专利]图像标注、获取图像标注模型的方法、装置及设备

权利要求书

1.一种图像标注的方法，其特征在于，所述方法包括：

获取第一图像，所述第一图像中存在至少一个待标注的目标对象；

获取所述第一图像的多个尺寸的特征图，所述多个尺寸的特征图中的各个尺寸的特征图包括多个第一特征向量；

从所述多个尺寸的特征图的第一特征向量中选择满足筛选条件的第一特征向量，得到至少一个目标对象对应的第一特征向量；

获取所述至少一个目标对象对应的第一特征向量之间的相关性数据，基于所述相关性数据标注所述至少一个目标对象中的各个目标对象。

2.根据权利要求1所述的方法，其特征在于，所述从所述多个尺寸的特征图的第一特征向量中选择满足筛选条件的第一特征向量，包括：

获取所述各个尺寸的特征图的关联评分，所述关联评分用于反应特征图中的多个第一特征向量与所述至少一个目标对象的关联度；

基于所述各个尺寸的特征图的关联评分从所述多个尺寸的特征图的第一特征向量中选择满足所述筛选条件的第一特征向量。

3.根据权利要求2所述的方法，其特征在于，所述基于所述各个尺寸的特征图的关联评分从所述多个尺寸的特征图的第一特征向量中选择满足所述筛选条件的第一特征向量，包括：

基于所述多个尺寸的特征图的第一特征向量的数量确定第一数量，将所述多个尺寸的特征图的第一特征向量按照关联度顺序排列，选择排列在所述第一数量之前的第一特征向量作为满足所述筛选条件的第一特征向量。

4.根据权利要求1所述的方法，其特征在于，所述基于所述相关性数据标注所述至少一个目标对象中的各个目标对象，包括：

基于所述相关性数据预测所述目标对象对应各个类别的概率；

基于所述目标对象对应各个类别的概率标注所述目标对象的类别。

5.根据权利要求1所述的方法，其特征在于，所述基于所述相关性数据标注所述至少一个目标对象中的各个目标对象，包括：

映射所述相关性数据得到所述目标对象的四维向量，基于所述四维向量标注所述目标对象的位置。

6.根据权利要求1-5任一所述的方法，其特征在于，所述方法还包括：

获取所述第一图像对应的图像标注模型，所述图像标注模型包括特征提取模块、筛选模块与自注意力模块；

所述获取所述第一图像的多个尺寸的特征图，包括：

调用所述特征提取模块获取所述第一图像的多个尺寸的特征图；

所述从所述多个尺寸的特征图的第一特征向量中选择满足筛选条件的第一特征向量，包括：

调用所述筛选模块从所述多个尺寸的特征图的第一特征向量中选择满足所述筛选条件的第一特征向量；

所述获取所述至少一个目标对象对应的第一特征向量之间的相关性数据，基于所述相关性数据标注所述至少一个目标对象中的各个目标对象，包括：

调用所述自注意力模块获取所述至少一个目标对象对应的第一特征向量之间的相关性数据，基于所述相关性数据标注所述至少一个目标对象中的各个目标对象。

7.根据权利要求6所述的方法，其特征在于，所述特征提取模块包括多层特征提取子模块，不同层的特征提取子模块输出的特征图的尺寸不同，上一层特征提取子模块输出的特征图为下一层特征提取子模块络的输入。