融合Shufflenet-V2的Yolov5轻量化目标检测方法  

作　　者：夏长权[1] 汪李超 韩一帆 徐思韵 时壮壮 陆文峰[1] XIA Changquan;WANG Lichao;HAN Yifan;XU Siyun;SHI Zhuangzhuang;LU Wenfeng

机构地区：[1]扬州大学物理科学与技术学院,江苏扬州225000

出　　处：《信息技术与信息化》2023年第3期100-104,共5页Information Technology and Informatization

基　　金：江苏省研究生实践创新计划,项目编号:SJCX21_1549。

摘　　要：选取目标检测领域中常见的城市道路车辆作为识别目标,针对如今日益增长的目标检测网络模型计算量与移动端设备匮乏的计算资源相冲突的问题,提出了一种将Shuffl enet-V2轻量级神经网络模型融合进Yolov5目标检测框架中的方法,修改网络头部卷积层,加入注意力机制SE-Layer层,并将激活函数修改为H-Swish函数。实验选取UA-DETRAC公开数据集,结果表明,模型大小达到了1.53 M,在同等平均准确率的模型中达到最低,为目标检测模型在计算资源有限的移动端设备中的应用与部署提供了有效的参考。

关 键 词：目标检测 Shuffl eNet-V2 轻量级神经网络 Yolov5 H-Swish函数 

分 类 号：TP391.41[自动化与计算机技术—计算机应用技术]