基于点云图卷积神经网络的3D目标检测
CSTR:
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

TP 391.4

基金项目:

国家自然科学基金资助项目(62276167);上海市自然科学基金资助项目(20ZR1437900)


3D object detection using graph convolutional neural network with point clouds
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    随着激光雷达传感器的快速发展,目标检测算法从传统的2D检测快速转向3D检测。然而,激光雷达产生的点云是不规则和非结构化的数据,传统的卷积神经网络无法对其进行处理。基于此提出了一种新颖的图卷积神经网络,能够更好地利用数据的几何关系和拓扑结构直接从点云中学习特征以进行3D目标检测。首先将原始激光雷达点云数据进行下采样,再进行固定半径邻域图的构建,随后设计了一个新型的图卷积神经网络对点云进行编码来预测图中每个顶点所属对象的类别和形状。为提升检测准确度,网络中加入了一种校准机制来减少特征在不同维度变化时引入的平移误差,此外还引入了注意力机制,以使用权重来进一步强化输出的顶点特征。在 KITTI 数据集上进行实验,实验结果表明,此方法能够有效对3D目标进行检测。对比其他多种检测算法,此方法在检测准确度上具有一定的优势。

    Abstract:

    With the rapid development of light detection and ranging (LiDAR) sensors, object detection algorithm moved quickly from traditional 2D detection to 3D detection. However, the point clouds generated by LiDAR were irregular and unstructured data, which cannot be adopted by conventional Convolutional Neural Networks. Thus, a novel Graph Convolutional Neural Network was proposed, which could better utilize the geometric relationship and topology of the data to learn features directly from point clouds for 3D object detection. First, the original LiDAR point cloud was randomly down-sampled and voxel down-sampled, and then constructed a fixed-radius neighborhood graph. Then, a novel graph convolutional neural network was designed to encode the point cloud to predict the class and shape of the object to which each vertex in the graph belongs. In order to improve the detection accuracy, we added a calibration mechanism to the network to reduce the translation error introduced by features changing in different dimensions, and also introduced an attention module to use weights to further strengthen the output vertex features. Finally, we conducted experiments on the KITTI dataset, and the experimental results showed that the method in this work could effectively detect 3D objects. Compared with many other detection algorithms, the method in this work had certain advantages in detection accuracy.

    参考文献
    相似文献
    引证文献
引用本文

刘振威,黄影平,梁振明,杨静怡.基于点云图卷积神经网络的3D目标检测[J].上海理工大学学报,2024,46(3):320-330.

复制
分享
相关视频

文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2023-01-15
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2024-07-12
  • 出版日期:
文章二维码