一种基于改进YOLOv8算法的遥感影像小目标识别方法

本申请公开了一种基于改进YOLOv8的遥感影像小目标识别方法，该方法包括：获取遥感影像数据集，进行预处理，生成预处理后的数据集；构建改进的YOLOv8模型，包括骨干网络、颈部网络和头部网络，其中头部网络包括小目标检测层，改进的YOLOv8模型采用EIoU损失函数衡量真实框和预测框之间的差异，采用回归加权的非极大抑制法对预测框位置进行加权平均，修正预测框的位置；基于预处理后的数据集对改进的YOLOv8模型进行训练，获取训练后的模型；基于训练后的模型对遥感影像进行小目标识别检测，输出检测结果。本发明还公开了一种基于改进YOLOv8的遥感影像小目标识别装置、相应的设备及存储介质。本发明能够有效减少遥感影像中小目标的漏检和误检现象，提升了检测性能。