节点文献

基于SimAM-YOLOv4的自动驾驶目标检测算法

Automatic driving target detection algorithm based on SimAM-YOLOv4

  • 推荐 CAJ下载
  • PDF下载
  • 不支持迅雷等下载工具,请取消加速工具后下载。

【作者】 刘丽伟; 侯德彪; 侯阿临; 梁超; 郑贺伟;

【Author】 LIU Liwei;HOU Debiao;HOU Alin;LIANG Chao;ZHENG Hewei;School of Computer Science and Engineering,Changchun University of Technology;

【通讯作者】 侯德彪;

【机构】 长春工业大学计算机科学与工程学院;

【摘要】 针对自动驾驶场景下单阶段目标检测对小目标精度不足的问题,权衡精度与速度的共同需求,提出一种改进的YOLOv4目标检测算法。首先,在网络的残差模块中嵌入SimAM注意力模块,旨在提高网络对重要特征的提取能力,然后,利用ACON-C激活函数替换残差模块中的Mish激活函数,使残差模块可以自适应地激活,进而提升网络性能。在KITTI数据集上进行训练和测试,实验结果 表明,该模型的平均精度均值达到91.16%,检测速度达到32帧/s,满足实时检测的要求。

【Abstract】 Aiming at the problem of insufficient precision for small targets in single-stage target detection in autonomous driving scenarios,and weighing the common requirements of accuracy and speed,an improved YOLOv4target detection algorithm is proposed.First,the SimAM attention module is embedded in the residual module of the network to improve the network’s ability to extract important features,and then the ACON-C activation function is used to replace the Mish activation function in the residual module,so that the residual module can adaptively activation,thereby improving the performance of the network.Training and testing are performed on the KITTI data set.The experimental results show that the average precision of the model reaches 91.16%,and the detection speed reaches 32FPS,which meets the requirements of real-time detection.

【关键词】 目标检测; YOLOv4; SimAM注意力模块; ACON-C;
【Key words】 object detection; YOLOv4; SimAM attention mechanism; ACON-C;
【基金】 吉林省科技发展计划重点项目(20200403037SF);吉林省教育厅科学技术研究规划项目(JJKH20210738KJ)
  • 【文献出处】 长春工业大学学报 ,Journal of Changchun University of Technology , 编辑部邮箱 ,2022年03期
  • 【分类号】TP183;TP391.41;U463.6
  • 【下载频次】57
节点文献中: 

本文链接的文献网络图示:

本文的引文网络