[发明专利]遥感图像建筑物的检测方法、装置、设备及存储介质

说明书

本申请提供一种遥感图像建筑物的检测方法、装置、设备及存储介质。通过接收待检测遥感图像，采用改进的特征提取模型对遥感图像进行多尺度特征提取，得到多个不同尺度的第一特征图；确定多个不同尺度的第一特征图中前景目标的第一检测框；获取第一检测框对应的第二特征图，对第二特征图进行分类、框回归以及掩膜预测，得到用于指示遥感图像中每种建筑物的类别信息、位置信息以及掩膜信息的检测结果。上述改进的特征提取模型包括多个卷积计算层，每个卷积计算层包括两个Ghost模块层，Ghost模块层是采用一系列线性运算生成特征图的模块层，对模型做了精度和轻量化上的改进，在满足高分遥感图像建筑物检测精度的情况下提升检测效率。

技术领域

本申请涉及计算机视觉技术领域，尤其涉及一种遥感图像建筑物的检测方法、装置、设备及存储介质。

背景技术

随着遥感卫星领域技术的不断发展，遥感影像中建筑物目标自动化精确提取成为数字测图领域重要的研究方向。高分遥感图像数据中存在大量的人工地物，部分人工地物的纹理、材料等特征细节与建筑物极度相似，且与目标建筑物地理位置较近，因此，这些干扰使得建筑物很难从背景中准确且完整地提取出来。

目前，基于深度学习的高分遥感图像建筑物模型大部分依靠语义分割方法或目标检测方法实现。随着遥感图像分辨率的提高，图像中每个像元包含的语义信息较为有限，建筑物占有的像元个数较多，这使得遥感图像中的建筑物的特征变得更加复杂难以学习和识别。

现有模型在数据处理过程有以下问题：提取的特征信息冗余，无法在高分辨率下精确分割建筑物，并且经常产生建筑物检测轮廓不完整的情况，使得高分遥感图像建筑物提取的精度不高。另外由于高分遥感图像数据量庞大，建筑物提取任务工作量巨大，提取特征效率低。

发明内容

本申请实施例提供一种遥感图像建筑物的检测方法、装置、设备及存储介质，在满足检测精度的情况下，提高检测效率。

本申请实施例的第一方面提供一种遥感图像建筑物的检测方法，包括：

接收待检测的遥感图像，所述遥感图像包括建筑物；

采用改进的特征提取模型对所述遥感图像进行多个不同尺度的特征提取，得到多个不同尺度的第一特征图；所述改进的特征提取模型包括多个卷积计算层，每个卷积计算层包括两个Ghost模块层，Ghost模块层是采用一系列线性运算生成特征图的模块层；

确定所述多个不同尺度的第一特征图中前景目标的第一检测框，所述前景目标包括所述建筑物以及除所述建筑物之外的其他目标；

获取所述第一检测框对应的第二特征图，对所述第二特征图进行分类、框回归以及掩膜预测，得到所述遥感图像的建筑物检测结果；所述建筑物检测结果用于指示所述遥感图像中每种建筑物的类别信息、位置信息以及掩膜信息。

本申请第一方面的一个可选实施例中，所述采用改进的特征提取模型对所述遥感图像进行多个不同尺度的特征提取，得到多个不同尺度的第一特征图，包括：

将遥感图像输入所述改进的特征提取模型中的主干网络，得到多个不同尺度的第三特征图；所述主干网络包括所述多个卷积计算层；

将所述多个不同尺度的第三特征图输入所述改进的特征提取模型中的特征融合网络，生成所述多个不同尺度的第一特征图。

本申请第一方面的一个可选实施例中，所述每个卷积计算层包括两个Ghost模块层，分别为第一Ghost模块和第二Ghost模块；

其中，所述第一Ghost模块层用于扩展输入的特征图的通道数，所述第二Ghost模块用于减少输入的特征图的通道数。

本申请第一方面的一个可选实施例中，所述第一Ghost模块的输入包括所述遥感图像，或者，所述第一Ghost模块对应的卷积计算层的前一个卷积计算层输出的特征图；

所述第二Ghost模块层的输入包括对所述第一Ghost模块层输出的特征图进行归一化BN和ReLU非线性激活后的特征图。