中文

Grid-GCN:用于快速可扩展点云学习的网格图卷积网络

计算机视觉与模式识别 2021-04-14 v5 机器学习

摘要

由于点云数据的稀疏性与不规则性,直接处理点的点的方法已变得流行。在所有基于点的模型中,图卷积网络(GCN)通过充分保留数据粒度并利用点间关联取得了显著性能。然而,基于点的网络在数据结构化(如最远点采样(FPS)与邻点查询)上耗费大量时间,限制了速度与可扩展性。本文提出一种名为 Grid-GCN 的方法,用于快速可扩展的点云学习。Grid-GCN 采用一种新颖的数据结构化策略——覆盖感知网格查询(CAGQ)。通过利用网格空间的高效性,CAGQ 在降低理论时间复杂度的同时改善了空间覆盖。与最远点采样(FPS)和球查询等常用采样方法相比,CAGQ 实现了高达 50 倍的加速。借助网格上下文聚合(GCA)模块,Grid-GCN 在主要的点云分类与分割基准上取得了最先进的性能,且运行时间显著快于以往研究。值得注意的是,Grid-GCN 在 ScanNet 上以每场景 81920 个点作为输入,实现了 50fps 的推理速度。

关键词

引用

@article{arxiv.1912.02984,
  title  = {Grid-GCN for Fast and Scalable Point Cloud Learning},
  author = {Qiangeng Xu and Xudong Sun and Cho-Ying Wu and Panqu Wang and Ulrich Neumann},
  journal= {arXiv preprint arXiv:1912.02984},
  year   = {2021}
}