首页|改进Yolov5s的移动端AR目标识别算法

改进Yolov5s的移动端AR目标识别算法

扫码查看
原文链接
  • 国家科技期刊平台
  • NETL
  • NSTL
  • 万方数据
针对目标识别模型存在参数量大、识别速度慢的问题,提出了一种改进的轻量化目标检测算法Yolov5s-MCB.将MobileNetV3网络作为Yolov5s主干特征提取网络以降低模型参数量.为了更好地拟合非线性数据优化模型收敛效果,将MobileNetV3网络ReLU激活函数替换成Mish激活函数以避免梯度消失和梯度爆炸.增加BiFPN特征金字塔结构,利用一种迭代式的特征融合方法提高检测精度.此外,引入坐标注意力机制使得模型关注大范围的位置信息以提高检测性能.为了优化模型训练收敛速度,采用Focal-Loss EIoU作为边框回归损失函数来解决低质量样本产生损失值剧烈震荡的问题.实验结果表明,该算法在VOC数据集的平均识别精度达到了90.5％,模型大小为7.63MB,检测速度为99FPS,与原Yolov5s相比,在保持识别精度不变的情况下,推理速度提升了17.85％,模型大小降低了45.9％,满足检测任务的实时性和检测精度要求.同时,将Yolov5s-MCB模型转为ONNX模型移植到手机上,结合ARCore SDK开发一个附带目标检测功能的AR应用.
Mobile AR target recognition algorithm based on improved Yolov5s

Yolov5slightweightattention mechanismmobile augmented reality

曹献烁、陈纯毅、胡小娟、于海洋、李延风

展开 >

长春理工大学 计算机科学技术学院, 长春 130022

Yolov5s 轻量化 注意力机制 移动增强现实

国家自然科学基金吉林省科技发展计划吉林省教育厅科研项目

U19A206320230201080GXJJKH20230851KJ

2023

10.3969/j.issn.1674-8425(z).2023.10.017

重庆理工大学学报
重庆理工大学

重庆理工大学学报

CSTPCD北大核心
影响因子：0.567
ISSN：1674-8425
年,卷(期)：2023.37(19)
  • 5