文档名：改进YOLOv5s的轻量化目标检测算法研究
摘要：针对当前YOLOv5s目标检测网络复杂、参数多、部署所需配置高,难以在嵌入式平台上获得优质识别结果的问题,设计了一种轻量化目标检测算法YOLOv5sGCB.算法使用GhostNet作为主干特征提取网络,充分发挥其计算量少、特征图不冗余的优势,从而降低算法的复杂度,提高检测速度;引入CA(coordinateattention)注意力机制,将空间坐标信息与注意力图有效整合,有助于网络快速提取有用特征,进一步增强算法的特征提取能力;借助双向特征金字塔网络(Bi-FPN)结构代替原始算法的路径聚合网络(pathaggregationnetwork)结构,对多个尺度的特征进行融合,以此构建新轻量化网络模型YOLOv5sGCB.与原始算法相比,改进后的算法在保持目标检测精确的同时精简了模型参数,降低了运行YOLOv5算法所需的硬件要求.在VOC2007数据集中,YOLOv5sGCB算法的平均准确率(mAP)达到75.2％,模型体积为10.6MB,浮点计算量11.3GFLOPs(gigafloating-pointoperationspersecond),与原始算法相比,参数量降低了30％,权重模型减少了20％.实验结果表明:YOLOv5sGCB算法在保证检测精确度的同时实现了模型的轻量化,为其在性能较弱的嵌入式平台上的部署与应用提供了一定的理论依据.
作者：龙邹荣 蔡林峰 叶彬强 汤斌 赵明富 唐跃林 王建旭 周密 Author：LONGZourong CAILinfeng YEBinqiang TANGBin ZHAOMingfu TANGYuelin WANGJianxu ZHOUMi
作者单位：重庆理工大学重庆市光纤传感与光电检测重点实验室,重庆400054重庆理工大学两江人工智能学院,重庆400054重庆市特种设备检测研究院,重庆401121;国家市场监管重点实验室,重庆401121
刊名：重庆理工大学学报 PKU
Journal：JournalofChongqingInstituteofTechnology
年，卷(期)：2023, 37(23)
分类号：TP391.4
关键词：轻量化网络 YOLOv5s 注意力机制 GhostNet 加权双向金字塔
Keywords：lightweightnetwork YOLOv5s attentionmechanism GhostNet weightedbidirectionalpyramid
机标分类号：TP391TP18TN91
在线出版日期：2024年3月1日
基金项目：国家市场监督管理总局科技计划，重庆市市场监督管理局科研计划项目，重庆市自然科学基金面上项目，重庆市教委科学研究计划项目，重庆市教委科学研究计划项目，重庆理工大学科研启动基金项目，重庆理工大学科研启动基金项目改进YOLOv5s的轻量化目标检测算法研究[
期刊论文] 重庆理工大学学报--2023, 37(23)龙邹荣 蔡林峰 叶彬强 汤斌 赵明富 唐跃林 王建旭 周密针对当前YOLOv5s目标检测网络复杂、参数多、部署所需配置高,难以在嵌入式平台上获得优质识别结果的问题,设计了一种轻量化目标检测算法YOLOv5sGCB.算法使用GhostNet作为主干特征提取网络,充分发挥其计算量少、特征图不冗...参考文献和引证文献
参考文献
引证文献
本文读者也读过
相似文献
相关博文
改进YOLOv5s的轻量化目标检测算法研究 Research on improved YOLOv5s lightweight target detection algorithm
改进YOLOv5s的轻量化目标检测算法研究.pdf
- 文件大小:
- 2.08 MB
- 下载次数:
- 60
-
高速下载
|
|
