摘要
本发明公开了一种无人机航拍下小目标的检测方法及检测系统,属于目标检测领域。包括:获取无人机航拍数据构建无人机航拍数据集;利用无人机航拍数据集对改进的YOLOv8模型进行训练,得到无人机航拍检测模型;将待检测的无人机航拍图像输入无人机航拍检测模型,获得无人机航拍检测结果。本发明在主干网络部分使用CDR模块替换P4、P5层的C2f模块,实现特征提取;在颈部引入BiFPN,利用跨尺度连接和加权融合输入特征,更高效实现多尺度特征融合;引入内容引导卷积和自注意力,通过学习到的空间权重调整输出特征图,缓解了梯度消失问题,增强了细节特征;在检测头部分引入动态检测头,实现检测头和注意力的统一,提高检测能力。
技术关键词
无人机航拍数据
多尺度特征融合
无人机航拍图像
双向特征金字塔
构建无人机
注意力
检测头
输出特征
网络
通道
全局平均池化
特征切片
级联
模型训练模块
洗牌
数据获取模块
动态