基于神经网络的遥感图像飞机实时检测算法

【类型】期刊

【作者】刘志  杨江涛  许新云 

【刊名】工业控制计算机

【关键词】 遥感图像,注意力机制,YOLOv4,MobileNetv3,深度可分离卷积

【摘要】针对在遥感图像飞机检测任务中精度较低、实时性差的问题，提出了一种改进YOLOv4遥感图像检测的算法。该算法采用轻量级网络MobileNetV3代替YOLOv4最初的特征提取网络，保证其特征提取能力情况下，减少参数量；同时在路聚合网络（PANet）中使用深度可分离卷积代替传统卷积；在主干网络中引入BAM注意力机制，提高整体模型的泛化能力。然后对NMS网络进行了优化，以提升模型的最终识别精度。最后在自建的遥感飞机数据集上进行训练和测试。实验结果表明：相比原YOLOv4算法，改进算法有着更高的检测精度和更快的检测速度。

【年份】2024

【作者单位】太原科技大学电子信息工程学院;

【期号】04

【页码】95-98