重庆理工大学学报(自然科学) ›› 2024, Vol. 38 ›› Issue (11): 35-42.

• 车辆工程 • 上一篇    下一篇

轻量级车辆行人检测模型研究及 Android部署

王道斌,李宸翔,严运兵   

  1. 武汉科技大学 汽车与交通工程学院,武汉 430065
  • 发布日期:2024-12-31
  • 作者简介:王道斌,男,博士,副教授,主要从事交通信息采集研究,Email:wangdaobin05@163.com;通信作者 李宸翔,男,硕士研究生,主要从事交通信息检测和计算机视觉研究,Email:837238173@qq.com;严运兵,男,博士,教授,主要从事智能驾驶人机交互与决策和新能源汽车驱动与控制研究

  • Published:2024-12-31

摘要: 针对目标检测模型参数量大,难以部署在移动端设备的问题,提出了一种轻量化车辆行人检测模型YOLOv8TI(trafficinformation)。设计全新的轻量级参数共享 SPGDetect检测头以降低模型的参数量和计算量;提出全局平衡通道路径聚合网络(GBCPAN)结构,平衡网络通道数量,通过跨尺度的加权链接,实现了自顶向下和自底向上的双向特征融合;此外,引入动态非单调聚焦机制的损失函数(WiseLoss)代替原损失函数以提升预测框精度。实验结果发现,提出的目标检测模型 YOLOv8TI在保持较高精度的同时,参数量、计算量和模型体积分别为YOLOv8n的 52.1%、58.0%和 54%。通过与其他轻量级目标检测模型对比,验证了该方法的有效性和卓越性。将YOLOv8TI进行 Android移动端部署,在荣耀 20和荣耀 80GT上进行了测试,FPS可达 24帧和 31帧,满足实时性需求,有望进一步集成在自动驾驶汽车上完成交通信息检测功能。

关键词: 深度学习;车辆行人检测;参数共享;轻量化

中图分类号: 

  • U491.1+13