arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2209.00860cs.CV
收藏

基于Transformer的三维单目标实时的跟踪

Real-time 3D Single Object Tracking with Transformer

Northeastern University(东北大学);Science and Technology on Near-Surface Detection Laboratory(近地面探测技术重点实验室)

Jiayao Shan, Sifan Zhou, Yubo Cui, Zheng Fang

摘要:

基于LiDAR的3D单目标跟踪是机器人和自动驾驶领域的一个具有挑战性的问题。目前,现有的方法通常遭受的问题,在长距离的对象往往是非常稀疏或部分闭塞的点云,这使得由模型提取的特征模糊。模糊特征会给目标定位带来困难,最终导致跟踪效果不佳。为了解决这个问题,我们利用强大的Transformer架构,并提出了一个点跟踪-Transformer(PTT)模块的点云为基础的3D单目标跟踪任务。具体而言,PTT模块通过计算注意力权重生成微调的注意力特征,引导跟踪器聚焦于目标的重要特征,提高复杂场景下的跟踪能力。为了评估我们的PTT模块,我们将PTT嵌入到主流方法中,并构建了一个新的3D SOT跟踪器PTT-Net。在PTT-Net中,我们将PTT分别嵌入到投票阶段和提案生成阶段。投票阶段的PTT模块可以模拟点块之间的交互,学习上下文相关的特征。同时,在建议生成阶段,PTT模块可以捕获对象和背景之间的上下文信息。我们在KITTI和NuScenes数据集上评估我们的PTT-Net。实验结果证明了PTT模块的有效性和PTT-Net的优越性,在汽车类别中超过了基线的明显幅度约10%。同时,我们的方法也有一个显着的性能改善稀疏场景。一般来说,Transformer和跟踪管道的组合使我们的PTT-Net能够在两个数据集上实现最先进的性能。此外,PTT-Net可以在NVIDIA 1080Ti GPU上以40 FPS实时的【形】运行。我们的代码在https://github.com/shanjiayao/PTT上对研究社区开放。

英文摘要:

LiDAR-based 3D single object tracking is a challenging issue in robotics and autonomous driving. Currently, existing approaches usually suffer from the problem that objects at long distance often have very sparse or partially-occluded point clouds, which makes the features extracted by the model ambiguous. Ambiguous features will make it hard to locate the target object and finally lead to bad tracking results. To solve this problem, we utilize the powerful Transformer architecture and propose a Point-Track-Transformer (PTT) module for point cloud-based 3D single object tracking task. Specifically, PTT module generates fine-tuned attention features by computing attention weights, which guides the tracker focusing on the important features of the target and improves the tracking ability in complex scenarios. To evaluate our PTT module, we embed PTT into the dominant method and construct a novel 3D SOT tracker named PTT-Net. In PTT-Net, we embed PTT into the voting stage and proposal generation stage, respectively. PTT module in the voting stage could model the interactions among point patches, which learns context-dependent features. Meanwhile, PTT module in the proposal generation stage could capture the contextual information between object and background. We evaluate our PTT-Net on KITTI and NuScenes datasets. Experimental results demonstrate the effectiveness of PTT module and the superiority of PTT-Net, which surpasses the baseline by a noticeable margin, ~10% in the Car category. Meanwhile, our method also has a significant performance improvement in sparse scenarios. In general, the combination of transformer and tracking pipeline enables our PTT-Net to achieve state-of-the-art performance on both two datasets. Additionally, PTT-Net could run in real-time at 40FPS on NVIDIA 1080Ti GPU. Our code is open-sourced for the research community at https://github.com/shanjiayao/PTT.

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

↑