基于改进YOLOv7-tiny的坦克车辆检测方法

作     者：郑陆石 胡晓锋 于伟国 赵东志 张鸿涛 ZHENG Lushi;HU Xiaofeng;YUWeiguo;ZHAO Dongzhi;ZHANG Hongtao

作者机构：中北大学机电工程学院太原030051 北方工程设计研究院有限公司石家庄050011 北方华安工业集团有限公司黑龙江齐齐哈尔161006 

基　　金：山西省技术基础科研项目(JSZL2019408B001) 

出 版 物：《兵器装备工程学报》 (Journal of Ordnance Equipment Engineering)

年 卷 期：2023年第44卷第12期

页      码：285-292页

摘      要：针对不同种类无人机航拍高度相差较大、图像分辨率不佳引起的坦克车辆检测算法效果不佳、速度慢等问题,提出一种基于改进YOLOv7-tiny的无人机视角坦克车辆检测算法。首先构建包含568幅图像、2132个目标的坦克车辆数据集。其次对YOLOv7-tiny网络进行3个方面改进:提出了AC-ELAN网络结构并加入3D注意力机制,提高对目标信息的提取能力;引入SPPCSPC结构进一步扩大模型的感受野,同时能够有效减少训练学习时间;将损失函数计算方法替换为WIoU,聚焦于普通质量锚框,加速了模型收敛。最后实验结果表明,改进算法在自建数据集上表现优异,比传统的YOLOv7-tiny平均精度提升5.0%,在GPU设备上检测速度达到71帧/s,能够在无人机计算平台实现实时检测。

主 题 词：目标检测 YOLOv7-tiny网络 非对称卷积 3D注意力机制 WIoU损失 

学科分类：13[艺术学] 08[工学] 09[农学] 1305[艺术学-设计学类] 083002[083002] 0830[工学-生物工程类] 081104[081104] 0826[工学-生物医学工程类] 0804[工学-材料学] 0903[农学-动物生产类] 0816[工学-纺织类] 081602[081602] 081101[081101] 081102[081102] 0811[工学-水利类] 

核心收录：

D　O　I：10.11809/bqzbgcxb2023.12.038

馆 藏 号：203125281...