基于改进YOLOv4的无人机目标检测方法
田港
张鹏
邹金霖
赵晓林
空军工程大学装备管理与无人机工程学院,西安,710051
摘要：针对无人机平台由于内存、算力有限而导致检测模型部署困难、检测速度降低的问题,提出了一种基于YOLOv4的改进模型.首先,为了减小模型内存占用、节省计算资源,根据目标尺寸特点,对YOLOv4原模型的预测层进行了改进,将三尺度检测模型改进为双尺度检测模型;其次,对双尺度检测模型进行正常训练,然后将其BN层的缩放因子进行稀疏训练,最后通过裁剪一定比例的通道数以再次减小模型内存占用提升检测速度.实验分析表明,在与原模型检测效果基本一样的情况下,最终改进模型的内存占用减少了60％,仅103 M,FPS提升了35％,达到了58帧/s.
关键词：无人机目标检测YOLOv4中小目标双尺度检测模型通道裁剪
分类号：V279(各类型航空器)TP391.4(计算技术、计算机技术)
资助基金：国家自然科学基金(61703422)
论文发表日期：2021-08-25
在线出版日期：2025-08-15（本平台首次上网日期，不代表文献的发表时间）
页数：6( 9-14 )
英文信息