一种基于ESR-YOLOv5的光学遥感图像目标检测方法

技术特征：
1.一种基于esr-yolov5的光学遥感图像目标检测方法，其特征在于，包括以下步骤：获取光学遥感影像数据集并输入至超分辨率生成对抗网络模型进行预处理，生成预处理后的光学遥感影像数据集；构建改进的yolov5网络模型；基于预处理后的光学遥感影像数据集对改进的yolov5网络模型进行训练，得到训练后的esr-yolov5模型；将待检测的遥感影像数据集输入至训练后的esr-yolov5模型，得到遥感影像目标检测结果。2.根据权利要求1所述一种基于esr-yolov5的光学遥感图像目标检测方法，其特征在于，所述获取光学遥感影像数据集并输入至超分辨率生成对抗网络模型进行预处理，生成预处理后的光学遥感影像数据集这一步骤，其具体包括：获取光学遥感影像数据集；通过双三次插值方法对光学遥感图像数据集进行缩小处理，得到低分辨率图像数据集并输入至超分辨率生成对抗网络模型；所述超分辨率生成对抗网络模型包括生成器和判别器；基于超分辨率生成对抗网络模型的生成器，对低分辨率图像数据集进行放大处理，生成初步的超分辨率图像数据集；基于超分辨率生成对抗网络模型的判别器，对初步的超分辨率图像数据集进行判别处理，生成超分辨率图像数据集；对低分辨率图像数据集和超分辨率图像数据集进行合并，得到预处理后的遥感影像数据集。3.根据权利要求2所述一种基于esr-yolov5的光学遥感图像目标检测方法，其特征在于，所述基于超分辨率生成对抗网络模型的判别器，对初步的超分辨率图像数据集进行判别处理，生成超分辨率图像数据集这一步骤，其具体包括：超分辨率生成对抗网络模型的判别器对超分辨率生成对抗网络模型的生成器生成的初步的超分辨率图像数据集与光学遥感影像数据集之间的的相似度进行计算与判断，直至满足预设的精度条件，输出超分辨率图像数据集。4.根据权利要求1所述一种基于esr-yolov5的光学遥感图像目标检测方法，其特征在于，所述构建改进的yolov5网络模型这一步骤，其具体包括：基于原始的yolov5网络模型，对原始的yolov5网络模型中锚框数值的长宽比进行重新设定处理，得到改进的yolov5网络模型。5.根据权利要求4所述一种基于esr-yolov5的光学遥感图像目标检测方法，其特征在于，所述基于预处理后的光学遥感影像数据集对改进的yolov5网络模型进行训练，得到训练后的esr-yolov5模型这一步骤，其具体包括：对预处理后的光学遥感影像数据集进行划分处理，得到训练集、验证集和测试集；基于训练集对改进的yolov5网络模型进行训练，得到遥感影像权重模型；基于验证集和测试集对遥感影像权重模型调参与测试，得到训练后的esr-yolov5模型。6.根据权利要求5所述一种基于esr-yolov5的光学遥感图像目标检测方法，其特征在
于，所述基于训练集对改进的yolov5网络模型进行训练，得到遥感影像权重模型这一步骤，其具体包括：将训练集输入至改进的yolov5网络模型进行训练，所述改进的yolov5网络模型包括输入层、下采样层、卷积层、csplayer层、sppbottleneck层、上采样层和输出层；基于改进的yolov5网络模型的输入层对训练集进行图像矩阵化处理，得到图像矩阵化后的训练集；基于改进的yolov5网络模型的下采样层对图像矩阵化后的训练集中的遥感影像尺寸进行降维和抽象处理，得到降维后的训练集；基于改进的yolov5网络模型的卷积层对降维后的训练集进行卷积核计算处理，得到训练集的特征信息；基于改进的yolov5网络模型的csplayer层对训练集的特征信息进行卷积计算与矩阵归一化处理，得到归一化后的训练集；基于改进的yolov5网络模型的sppbottleneck层对归一化后的训练集进行卷积核计算处理，得到计算结果；基于改进的yolov5网络模型的上采样层对计算结果中的遥感影像的尺寸进行还原处理，得到矩阵化遥感影像；基于改进的yolov5网络模型的输出层将矩阵化遥感影像输出，构建遥感影像权重模型。7.根据权利要求5所述一种基于esr-yolov5的光学遥感图像目标检测方法，其特征在于，所述基于验证集和测试集对遥感影像权重模型调参与测试，得到训练后的esr-yolov5模型这一步骤，其具体包括：将验证集输入至遥感影像权重模型进行验证处理，得到初步的遥感影像目标检测结果；根据初步的遥感影像目标检测结果对遥感影像权重模型的超参数进行调整处理，得到调整后的遥感影像权重模型；将测试集输入至调整后的遥感影像权重模型进行测试，得到训练后的esr-yolov5模型。

技术总结
本发明公开了一种基于ESR-YOLOv5的光学遥感图像目标检测方法，该方法包括：获取光学遥感影像数据集并输入至超分辨率生成对抗网络模型进行预处理，生成预处理后的光学遥感影像数据集；构建改进的YOLOv5网络模型；基于预处理后的光学遥感影像数据集对改进的YOLOv5网络模型进行训练，得到训练后的ESR-YOLOv5模型；将待检测的遥感影像数据集输入至训练后的ESR-YOLOv5模型，得到遥感影像目标检测结果。通过使用本发明，不仅能够对光学遥感图像的质量进行了提高还能够提高ESR-YOLOv5网络整体的检测精度和普适性的程度。本发明作为一种基于ESR-YOLOv5的光学遥感图像目标检测方法，可广泛应用于光学遥感图像处理技术领域。广泛应用于光学遥感图像处理技术领域。广泛应用于光学遥感图像处理技术领域。


技术研发人员：方坤 黄旭光
受保护的技术使用者：华南师范大学
技术研发日：2022.09.02
技术公布日：2022/11/11