arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2007.12392cs.CVcs.LGeess.IV
收藏

基于LSTM的LiDAR点云三维目标检测方法

An LSTM Approach to Temporal 3D Object Detection in LiDAR Point Clouds

Google Research(谷歌研究院)

Rui Huang, Wanyue Zhang, Abhijit Kundu, Caroline Pantofaru, David A Ross, Thomas Funkhouser, Alireza Fathi

摘要:

在3D LiDAR数据中检测物体是自动驾驶和其他机器人应用的核心技术。虽然LiDAR数据是随时间采集的,但大多数3D目标检测算法针对每个帧独立地提出对象边界框,而忽略了时域中可用的有用信息。为了解决这个问题,在本文中,我们提出了一种基于稀疏LSTM的多帧三维目标检测算法。我们使用U-Net风格的3D稀疏卷积网络来提取每帧LiDAR点云的特征。这些特征与上一帧的隐藏和记忆特征一起被馈送到LSTM模块,以预测当前帧中的3d对象以及传递到下一帧的隐藏和记忆特征。在Waymo开放数据集上的实验表明,我们的算法比传统的逐帧方法提高了7.5%mAP@0.7,比其他多帧方法提高了1.2%,同时每帧使用更少的内存和计算量。据我们所知,这是第一个使用LSTM在稀疏点云中进行3D目标检测的工作。

英文摘要:

Detecting objects in 3D LiDAR data is a core technology for autonomous driving and other robotics applications. Although LiDAR data is acquired over time, most of the 3D object detection algorithms propose object bounding boxes independently for each frame and neglect the useful information available in the temporal domain. To address this problem, in this paper we propose a sparse LSTM-based multi-frame 3d object detection algorithm. We use a U-Net style 3D sparse convolution network to extract features for each frame's LiDAR point-cloud. These features are fed to the LSTM module together with the hidden and memory features from last frame to predict the 3d objects in the current frame as well as hidden and memory features that are passed to the next frame. Experiments on the Waymo Open Dataset show that our algorithm outperforms the traditional frame by frame approach by 7.5% mAP@0.7 and other multi-frame approaches by 1.2% while using less memory and computation per frame. To the best of our knowledge, this is the first work to use an LSTM for 3D object detection in sparse point clouds.

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

↑