基于注意力机制的航拍图像地面目标旋转检测实验设计
2024-05-31分类号:TP391.41
【部门】北京理工大学航天学院 北京科技大学自动化学院
【摘要】针对无人机航拍图像中目标实例存在着遮挡、旋转、尺度变化等情况,该文设计了一种基于注意力机制的旋转目标检测实验设计方案。首先,对开源DOTAv-1.0数据集样本图像进行了图像增强和裁剪,构造二次数据集,使整体实验样本数量提升11.72倍。然后,通过在S~(2)A-Net网络的特征融合网络部分嵌入CBAM注意力机制模块,增强对输入特征空间和通道的有效信息利用率。最后,通过消融实验验证了CBAM模块的有效性,整体性能提升1.3%,对航拍小目标实例small-vehicle的检测精度mAP值由0.402 9提升到0.539 8,提升了34%,检测速率达到24 fps。
【关键词】深度学习 航空遥感 倾斜检测 注意力机制 实验设计
【基金】
【所属期刊栏目】实验技术与管理
文献传递