基于YOLOv5改进的轻量化目标检测

摘要：

针对移动端目标检测算法需要模型参数量与计算量更少、推理速度更快和检测效果更好以及目标检测算法对于小目标误检、漏检及特征提取能力不足等问题,提出一种基于YOLOv5改进的轻量化目标检测算法.该算法使用轻量级网络MobileNetV2作为目标检测算法的骨干网络降低模型的参数量与计算量,通过使用深度可分离卷积结合大卷积核的思想降低网络的计算量与参数量,并提升了小目标的检测精度.使用GhostConv来替换部分普通卷积,进一步降低参数量与计算量.本文算法在VOC竞赛数据集,COCO竞赛数据集两份数据集上均进行了多次对比实验,结果表明本文算法相比于其他模型参数量更小、计算量更小、推理速度更快以及检测精度更高.

关键词：轻量化、深度学习、特征金字塔网络(FPN)、YOLOv5、大核卷积

所属期刊栏目：32

分类号：TP391;TN911.73;TP193

资助基金：广东省基础与应用基础研究基金;广东省基础与应用基础研究基金;广东省普通高校重点专项;茂名市科技计划;广东石油化工学院博士启动项目

在线出版日期：2023-09-25（万方平台首次上网日期，不代表论文的发表时间）

页数：共11页

页码：132-142