基于空频特征融合检测头的无人机目标检测方法
摘要:
本发明公开了基于空频特征融合检测头的无人机目标检测方法,具体包括如下步骤方法:步骤1,输入公开数据集,并划分为训练集和测试集;步骤2,对训练集和测试集中的图片进行预处理,得到图片X;步骤3,构建无人机场景下基于空频特征融合检测头的YOLO目标检测网络模型;步骤4,对步骤3构建的无人机场景下空频特征融合检测头的YOLO目标检测网络模型进行训练;步骤5,将步骤2预处理后的测试集图片放入步骤4训练好的模型,最终得到带有检测框的无人机图像。本发明通过级联注意力增加全局信息提高了在背景复杂、场景拥挤下的检测能力,通过空频特征融合检测头解决尺度变化大的检测难题并提高了整体的检测性能。
0/0