arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2003.01251cs.CV
收藏

Point-GNN:用于点云中三维目标检测的图神经网络

Point-GNN: Graph Neural Network for 3D Object Detection in a Point Cloud

Carnegie Mellon University(卡内基梅隆大学)

Weijing Shi, Ragunathan, Rajkumar

摘要:

在本文中,我们提出了一个图神经网络检测对象从激光雷达点云。为此,我们在一个固定半径的近邻图中有效地编码点云。我们设计了一个图神经网络,称为Point-GNN,用于预测图中每个顶点所属对象的类别和形状。在Point-GNN中,我们提出了一种自动配准机制来减少平移方差,并设计了一种框合并和评分操作来准确地组合来自多个顶点的检测。我们在KITTI基准测试上的实验表明,该方法仅使用点云就达到了领先的精度,甚至可以超过基于融合的算法。我们的研究结果表明,潜在的使用图神经网络作为一种新的方法,三维目标检测。代码可在https://github.com/WeijingShi/Point-GNN上获得。

英文摘要:

In this paper, we propose a graph neural network to detect objects from a LiDAR point cloud. Towards this end, we encode the point cloud efficiently in a fixed radius near-neighbors graph. We design a graph neural network, named Point-GNN, to predict the category and shape of the object that each vertex in the graph belongs to. In Point-GNN, we propose an auto-registration mechanism to reduce translation variance, and also design a box merging and scoring operation to combine detections from multiple vertices accurately. Our experiments on the KITTI benchmark show the proposed approach achieves leading accuracy using the point cloud alone and can even surpass fusion-based algorithms. Our results demonstrate the potential of using the graph neural network as a new approach for 3D object detection. The code is available https://github.com/WeijingShi/Point-GNN.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑