|国家科技期刊平台
首页|期刊导航|无线电工程|改进YOLOv5的军事飞机检测算法

改进YOLOv5的军事飞机检测算法OA

中文摘要

针对遥感图像中军事飞机目标检测存在的精度低、漏检和虚警率高等问题,提出了一种基于YOLOv5s的轻量化遥感图像军事飞机目标检测算法——YOLO-Military Aircraft Recognition(YOLO-MAR)。提出新的网络结构,完成多尺度感受野权重调整,重设计特征提取网络和特征融合网络,实现小目标特征权重增加,并进行轻量化处理;使用FPGM对重构后的模型进行剪枝,极大地降低了模型的参数量和体积;使用SIoU Loss作为模型的损失函数,使模型的收敛速度加快并提升检测的精度。结果表明,在公开军用飞机数据集MAR20上,YOLO-MAR相比于原YOLOv5s,模型体积降低至3.95 MB,缩小了71.5%,经过剪枝后的模型体积最小可缩减至0.2 MB,模型平均检测精度最高可达91.7%,提高了2.34%,并且在检测效果、模型体积、参数量和计算量等方面具有先进性,能够对军用飞机目标进行高质量实时检测。

王杰;张上;张岳;胡益民;

三峡大学水电工程智能视觉监测湖北省重点实验室,湖北宜昌443002 三峡大学湖北省建筑质量检测装备工程技术研究中心,湖北宜昌443002 三峡大学计算机与信息学院,湖北宜昌443002

目标检测军事飞机YOLOv5sFPGMSIoU Loss

《无线电工程》 2024 (003)

P.589-596 / 8

国家级大学生创新创业训练计划(202111075012,202011075013)~~。

10.3969/j.issn.1003-3106.2024.03.010

评论