重庆理工大学学报(自然科学) ›› 2024, Vol. 38 ›› Issue (9): 143-149.

• 信息·计算机 • 上一篇    

一种针对室内关键目标检测的改进 YOLOv8算法

岳有军 ,张远锟,赵 辉,王红君   

  1. 1.天津理工大学 电气工程与自动化学院,天津 300384;2.天津市复杂系统控制理论及应用重点实验室,天津 30038
  • 发布日期:2024-10-12
  • 作者简介:岳有军,男,博士,教授,主要从事复杂系统建模及智能控制、机器人导航与控制技术、电力电子技术及应用研究,Email:bakeryueyj@163.com;通信作者 张远锟,男,硕士研究生,主要从事计算机视觉研究,Email:tutzhangyuankun13@163.com。

  • Published:2024-10-12

摘要: 随着社会服务型机器人的发展,室内目标检测成为机器人识别场景的重要任务。针对现有网络在室内目标检测任务中存在的检测精度低、检测速度慢、难以应用在嵌入式设备上的问题,提出了一种基于 YOLOv8的轻量化改进网络。针对场景中存在难以识别的小目标的问题,增加一个检测头以提高对小目标的检测精度;引入 GhostBottleneck替换网络 Neck部分中 C2f模块中的 bottlencek,将网络中后半部分卷积中的 SiLu激活函数替换为 Hswish激活函数,减少网络的参数量和计算量,提高检测速度,降低网络的移植难度;在 Neck部分中添加 MRLA注意力机制,加强不同层之间的联系,增加特征提取能力,提高整体识别精度。实验结果表明:在室内场景数据集上,改进后的算法较原算法平均精度提升了 3.6%,检测速度为 72frame/s,同时网络参数量较原网络减少约 11%,能满足检测的准确性和实时性,优于目前主流算法。

关键词: 目标检测;YOLOv8;室内场景;注意力机制;GhostNe

中图分类号: 

  • TP391.4