重庆理工大学学报(自然科学) ›› 2024, Vol. 38 ›› Issue (10): 55-62.

• 车辆工程 • 上一篇    

基于改进 PointPillars的 3D目标检测算法

谢生龙,邵金菊,单少飞,孙福昌,王 磊   

  1. 1.山东理工大学 交通与车辆工程学院,山东 淄博 255000;2.睿羿科技(山东)有限公司,山东 淄博 255000
  • 发布日期:2024-11-08
  • 作者简介:谢生龙,男,硕士,主要从事自动驾驶环境感知研究,Email:18204780610@163.com;通信作者 邵金菊,女,博士,副教授,主要从事自动驾驶关键技术研究,Email:sjjgbh@163.com。

  • Published:2024-11-08

摘要: 针对自动驾驶场景下远距离及遮挡目标识别问题,对 PointPillars算法进行了改进。引入了并行的空间注意力和通道注意力机制,增强了目标的位置信息及有用特征通道权重,提高了远距离目标的检测精度。在 2DCNN骨干网络中引入了自适应空间特征融合模块,解决了特征拼接的信息丢失问题,提高了遮挡目标的检测精度。基于KITTI数据集在 3种不同场景难度下分别对 SECOND、PointPillars、改进 PointPillars这 3种算法进行了定量分析验证,并将改进的 PointPillars算法进行可视化分析。定量分析表明,改进 PointPillars算法在鸟瞰图模式下目标检测精度最大提升 2.75%;在三维模式下目标检测精度最大提升 2.93%;在 AOS模式下目标检测精度最大提升 4.05%,可视化结果表明改进的 PointPillars算法能有效检测远距离及遮挡目标。

关键词: 目标检测;PointPillars;注意力机制;点云;自适应空间特征融合

中图分类号: 

  • TP18