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

三维点云目标检测中的注意力机制研究

Investigating Attention Mechanism in 3D Point Cloud Object Detection

Australian National University(澳大利亚国立大学);Data61-CSIRO;University of Technology Sydney(悉尼科技大学);Nanyang Technological University(南洋理工大学)

Shi Qiu, Yunfan Wu, Saeed Anwar, Chongyi Li

摘要:

三维(3D)空间中的目标检测吸引了学术界和工业界的极大兴趣,因为它是人工智能驱动的应用(诸如机器人、自动驾驶和增强现实)中的基本任务。点云作为三维数据的基本格式,能够提供原始三维空间中物体的详细几何信息。然而,由于3D数据的稀疏性和无序性,需要专门设计的网络和模块来处理这类数据。注意力机制在不同的计算机视觉任务中取得了令人印象深刻的性能,然而,目前还不清楚注意力模块将如何影响3D点云目标检测的性能,以及什么样的注意力模块可以适合3D数据的固有属性。这项工作研究的注意力机制在三维点云目标检测的作用,并提供洞察不同的注意力模块的潜力。为了实现这一目标,我们全面研究了经典的2D注意力,新颖的3D注意力,包括SUN RGB-D和ScanNetV 2数据集上的最新点云Transformers。在详细的实验和分析的基础上,我们总结了不同的注意模块的效果。希望本文的研究成果能为关注嵌入式三维点云目标检测提供参考。代码和训练模型可在https://github.com/ShiQiu0419/attentions_in_3D_detection上获得。

英文摘要:

Object detection in three-dimensional (3D) space attracts much interest from academia and industry since it is an essential task in AI-driven applications such as robotics, autonomous driving, and augmented reality. As the basic format of 3D data, the point cloud can provide detailed geometric information about the objects in the original 3D space. However, due to 3D data's sparsity and unorderedness, specially designed networks and modules are needed to process this type of data. Attention mechanism has achieved impressive performance in diverse computer vision tasks; however, it is unclear how attention modules would affect the performance of 3D point cloud object detection and what sort of attention modules could fit with the inherent properties of 3D data. This work investigates the role of the attention mechanism in 3D point cloud object detection and provides insights into the potential of different attention modules. To achieve that, we comprehensively investigate classical 2D attentions, novel 3D attentions, including the latest point cloud transformers on SUN RGB-D and ScanNetV2 datasets. Based on the detailed experiments and analysis, we conclude the effects of different attention modules. This paper is expected to serve as a reference source for benefiting attention-embedded 3D point cloud object detection. The code and trained models are available at: https://github.com/ShiQiu0419/attentions_in_3D_detection.

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

↑