[发明专利]一种基于深度神经网络的遥感图像目标检测方法

摘要

一种基于深度神经网络的遥感图像目标检测方法，包括：获取训练样本并对所述训练样本进行缩放预处理操作；在预处理后的所述训练样本中，对包含遥感目标及遥感背景的训练样本图像执行多种类型的标记操作；基于最大迭代次数，对选取的深度卷积神经网络模型执行预训练和微调操作，得到微调的深度卷积神经网络模型；对所述微调的深度卷积神经网络模型的全连接层进行处理，得到微调的深度全卷积网络模型；基于所述微调的深度卷积神经网络模型及所述微调的深度全卷积网络模型对遥感图像目标进行检测。该方法具有检测速度快、检测精度高等优点，可被广泛应用于目标检测、目标跟踪、智能导航、城市规划等领域。

主权项

1.一种基于深度神经网络的遥感图像目标检测方法，其特征在于，包括：获取训练样本并对所述训练样本进行缩放预处理操作；在预处理后的所述训练样本中，对包含遥感目标及遥感背景的训练样本图像执行多种类型的标记操作；基于最大迭代次数，对选取的深度卷积神经网络模型执行预训练和微调操作，得到微调的深度卷积神经网络模型；对所述微调的深度卷积神经网络模型的全连接层进行处理，得到微调的深度全卷积网络模型；基于所述微调的深度卷积神经网络模型及所述微调的深度全卷积网络模型对遥感图像目标进行检测，具体包括：(1)获取测试样本从遥感图像数据集中选取5～50幅测试样本图像，测试样本图像不要求范围和分辨率，也不限制大小和形状，所有测试样本图像不需要进行额外的预处理；(2)整合检测模型训练检测模型的步骤全部完成后，整合完成微调的深度卷积神经网络模型、完成微调的深度全卷积网络模型、滑动窗口算法和检测框融合算法，得到可用于遥感图像目标检测的检测模型，检测模型可以完成提取候选区域、候选区域的特征提取和分类以及检测框融合的功能；(3)提取候选区域(a)把测试样本图像作为输入图像输入检测模型，检测模型首先使用完成微调的深度全卷积网络模型处理输入图像，完成微调的深度全卷积网络模型的输出是和输入图像规模相同的分类图，输入图像中可能是遥感目标的像素点在分类图上的值为1，输入图像中可能是遥感背景的像素点在分类图上的值为0；(b)使用滑动窗口算法得出候选区域，使用滑动窗口算法之前需要设定窗口大小、窗口步长和提取阈值，滑动窗口算法以设定的窗口大小和窗口步长在分类图上进行从左上角到右下角的遍历搜索，如果窗口中的值为1的像素点的占比高于提取阈值，则该窗口被提取为候选区域；(4)候选区域的特征提取和分类把所有候选区域输入完成微调的深度卷积神经网络模型进行候选区域的特征提取和分类，需要提前设定分类阈值，完成微调的深度卷积神经网络模型的输出是把候选区域分类为遥感目标和遥感背景的置信度，把候选区域分类为遥感目标的置信度大于分类阈值时，用红色边框在原图中标记出该区域，得到检测框；(5)检测框融合设定融合阈值，使用检测框融合算法处理一幅测试样本图像中的所有检测框，可以得到最终的遥感图像目标检测结果。