基于YOLOv8的光学遥感图像舰船检测方法及装置
摘要:
本发明提供基于YOLOv8的光学遥感图像舰船检测方法及装置,涉及遥感图像识别技术领域。此方法包括:构建智能舰船检测模型,对第一数据集和第二数据集分别预处理,得到具有标注的第一舰船图像和第二舰船图像;将第一舰船图像和第二舰船图像依次输入智能舰船检测模型,对智能舰船检测模型进行训练,得到训练好的智能舰船检测模型;将预测图像输入训练好的智能舰船检测模型,检测出预测舰船信息;对预测舰船信息进行后处理,得到具有标注的检测图像。智能舰船检测模型的Backbone模块包括LSKblock、引入DWR模块的C2f模块,使特征图的特征表示全面,且引入SimAM的Neck模块,使对微小密集舰船目标的检测精度较高。
0/0