文档名：改进Yolov5s的移动端AR目标识别算法
摘要：针对目标识别模型存在参数量大、识别速度慢的问题,提出了一种改进的轻量化目标检测算法Yolov5s-MCB.将MobileNetV3网络作为Yolov5s主干特征提取网络以降低模型参数量.为了更好地拟合非线性数据优化模型收敛效果,将MobileNetV3网络ReLU激活函数替换成Mish激活函数以避免梯度消失和梯度爆炸.增加BiFPN特征金字塔结构,利用一种迭代式的特征融合方法提高检测精度.此外,引入坐标注意力机制使得模型关注大范围的位置信息以提高检测性能.为了优化模型训练收敛速度,采用Focal-LossEIoU作为边框回归损失函数来解决低质量样本产生损失值剧烈震荡的问题.实验结果表明,该算法在VOC数据集的平均识别精度达到了90.5％,模型大小为7.63MB,检测速度为99FPS,与原Yolov5s相比,在保持识别精度不变的情况下,推理速度提升了17.85％,模型大小降低了45.9％,满足检测任务的实时性和检测精度要求.同时,将Yolov5s-MCB模型转为ONNX模型移植到手机上,结合ARCoreSDK开发一个附带目标检测功能的AR应用.
作者：曹献烁 陈纯毅 胡小娟 于海洋 李延风Author：CAOXianshuo CHENChunyi HUXiaojuan YUHaiyang LIYanfeng
作者单位：长春理工大学计算机科学技术学院,长春130022
刊名：重庆理工大学学报 PKU
Journal：JournalofChongqingInstituteofTechnology
年，卷(期)：2023, 37(19)
分类号：TP391
关键词：Yolov5s 轻量化 注意力机制 移动增强现实
Keywords：Yolov5s lightweight attentionmechanism mobileaugmentedreality
机标分类号：TP391.41TN957.52TP183
在线出版日期：2023年12月4日
基金项目：国家自然科学基金，吉林省科技发展计划项目，吉林省教育厅科学研究项目改进Yolov5s的移动端AR目标识别算法[
期刊论文] 重庆理工大学学报--2023, 37(19)曹献烁 陈纯毅 胡小娟 于海洋 李延风针对目标识别模型存在参数量大、识别速度慢的问题,提出了一种改进的轻量化目标检测算法Yolov5s-MCB.将MobileNetV3网络作为Yolov5s主干特征提取网络以降低模型参数量.为了更好地拟合非线性数据优化模型收敛效果,将Mobil...参考文献和引证文献
参考文献
引证文献
本文读者也读过
相似文献
相关博文
改进Yolov5s的移动端AR目标识别算法 Mobile AR target recognition algorithm based on improved Yolov5s
改进Yolov5s的移动端AR目标识别算法.pdf
- 文件大小:
- 5.58 MB
- 下载次数:
- 60
-
高速下载
|
|
