苏州科技大学学报(自然科学版)2024,Vol.41Issue(2) :63-70.DOI:10.12084/j.issn.2096-3289.2024.02.010

基于深度学习的自动驾驶场景3D目标检测方法

Deep learning based 3D target detection method for autonomous driving scenes

张学锋 唐永吉 杨武洲 樊旭 黄永鹤 谢悦
苏州科技大学学报(自然科学版)2024,Vol.41Issue(2) :63-70.DOI:10.12084/j.issn.2096-3289.2024.02.010

基于深度学习的自动驾驶场景3D目标检测方法

Deep learning based 3D target detection method for autonomous driving scenes

张学锋 1唐永吉 1杨武洲 2樊旭 1黄永鹤 1谢悦1
扫码查看

作者信息

  • 1. 安徽工业大学特种重载机器人安徽省重点实验室,安徽马鞍山 243032
  • 2. 马鞍山市消防救援支队,安徽马鞍山 243032
  • 折叠

摘要

针对传统PV-RCNN在点云上采样效率低下和采样精度存在偏差等问题,提出了一种基于PV-RCNN改进的3D目标检测方法.更改关键点采样策略,使得有限的关键点可以更加地聚集在proposal区域范围内,更多的编码有效前景点特征来用于后面的proposal refinement,有效产生更具有代表性的关键点.用局部特征聚合的VectorPool聚合模块取代体素集抽象和ROI网格池化模块中的集合抽象,更高效的针对稀疏和不规则点云数据进行编码.在KITTI数据集上对算法验证,结果表明:行人鸟瞰图检测,困难级别检测精度提升较为显著,达到了 10.46%,整体帧率提升为33.74%,文中的方法拥有更好的检测性能.

Abstract

Aiming at the problems of low sampling efficiency and biased sampling accuracy of traditional PV-RCNN on the point cloud,an improved 3D target detection method based on PV-RCNN is proposed.The key point sampling strategy has been changed so that the limited key points can be more clustered within the scope of the proposal region and more effective foreground features are encoded to be used in the later proposal refine-ment,effectively generating more representative key points.We replaced the voxel set abstraction and the set ab-straction in the ROI grid pooling module with the VectorPool aggregation module for local feature aggregation to encode sparse and irregular point cloud data more efficiently.The algorithm is validated on the KITTI dataset,and the results show that the pedestrian bird's eye view detection takes a more significant difficulty level im-provement of 10.46%,and an overall frame rate improvement of 33.74%.Our method has better detection perfor-mance.

关键词

3D目标检测/卷积神经网络/点云/SPC关键点采样/VectorPool聚合模块

Key words

3D target detection/convolutional neural network/point clouds/SPC key point sampling/VectorPool aggregation module

引用本文复制引用

基金项目

安徽省高校自然科学研究项目(2022AH050290)

特种重载机器人安徽省重点实验室开放课题(TZJQR007-2023)

出版年

2024
苏州科技大学学报(自然科学版)
苏州科技学院

苏州科技大学学报(自然科学版)

影响因子:0.185
ISSN:2096-3289
参考文献量6
段落导航相关论文