[发明专利]用于识别图像的方法和装置

说明书

本公开的实施例公开了云计算中用于识别图像的方法和装置。该方法的一具体实施方式包括：将目标遥感图像划分成子图，得到子图集合；将子图集合中各个子图分别输入预先训练的目标检测模型，得到各个子图对应的目标区域，存入目标区域集合，其中，目标检测模型基于各子图的多层特征图检测各子图的目标区域，多层特征图中的各层特征图所描述的子图的特征不同；基于目标区域集合中的各个目标区域，确定各个目标区域所呈现的目标的身份信息；以及在目标遥感图像上，在对应各个目标区域的坐标位置处，标示对应各个目标区域所呈现的目标的身份信息。该实施方式可以实现对遥感图像中的小目标进行检测与识别。

技术领域

本公开的实施例涉及计算机技术领域，具体涉及用于识别图像的方法和装置。

背景技术

随着各种越来越高空间、时间、光谱分辨率卫星发射升空，基于卫星的遥感图像开始被广泛应用到军事领域和民用领域。遥感图像通常尺寸很大，且受分辨率制约，遥感图像中所呈现的物体的图像的尺寸往往很小。

因此，相关技术中，需要从尺寸大的遥感图像中检测出相对尺寸小的目标物体所在区域，以及对目标物体所在区域呈现的目标物体进行识别。

发明内容

本公开的实施例提出了用于识别图像的方法和装置。

第一方面，本公开的实施例提供了一种用于识别图像的方法，该方法包括：将目标遥感图像划分成子图，得到子图集合；将子图集合中各个子图分别输入预先训练的目标检测模型，得到各个子图对应的目标区域，存入目标区域集合，其中，目标检测模型基于各子图的多层特征图检测各子图的目标区域，多层特征图中的各层特征图所描述的子图的特征不同；基于目标区域集合中的各个目标区域，确定各个目标区域所呈现的目标的身份信息；以及在目标遥感图像上，在对应各个目标区域的坐标位置处，标示对应各个目标区域所呈现的目标的身份信息。

在一些实施例中，预先训练的目标检测模型被配置成执行以下步骤：提取所输入的子图的预设数目个特征图，从所提取的预设数目个特征图中确定出尺寸最大的特征图，以及提取尺寸最大的特征图的尺寸；将预设数目个特征图中除尺寸最大的特征图外的特征图的尺寸调整为与尺寸最大的特征图的尺寸相同；对预设数目个相同尺寸的特征图进行卷积运算，得到新特征图，以及基于新特征图确定出所输入的子图的目标区域。

在一些实施例中，基于目标区域集合中的各个目标区域，确定各个目标区域所呈现的目标的身份信息包括：将目标区域集合中的各个目标区域分别输入预先训练的细粒度图像分类模型，得到各个目标区域所呈现的目标的身份信息。

在一些实施例中，将目标遥感图像划分成子图，得到子图集合包括：按照预设偏移顺序和偏移重叠量，将目标遥感图像划分为子图，得到子图集合，其中，子图的尺寸为预设参考尺寸。

在一些实施例中，方法还包括：获取遥感图像，对遥感图像进行图像预处理，得到目标遥感图像。

在一些实施例中，对遥感图像进行图像预处理，包括以下至少一项：从遥感图像中识别出预设参照物，获取预设参照物在遥感图像中的位置信息，以及基于位置信息，确定遥感图像在预设坐标系中的角度和位移，按照角度和位移，调整遥感图像；将遥感图像的分辨率提升为预设分辨率。

在一些实施例中，将遥感图像的分辨率提升为预设分辨率，包括：将遥感图像输入预先训练的分辨率调整模型，以得到具有预设分辨率的遥感图像，其中，分辨率调整模型为生成对抗网络模型。

在一些实施例中，预先训练的目标检测模型，通过如下方式训练得到：获取第一训练样本集合，其中，第一训练样本包括子图和子图的目标区域以及目标区域的旋转角度；将第一训练样本集合中的第一训练样本的子图作为输入，将与输入的子图对应的目标区域以及目标区域的旋转角度作为期望输出，训练得到目标检测模型。