中文

Object DGCNN:使用动态图的 3D 目标检测

计算机视觉与模式识别 2021-10-14 v1 人工智能 机器学习 机器人学

摘要

3D 目标检测通常涉及复杂的训练和测试流程,需要对各个数据集的大量领域知识。受近期无抑制非极大值抑制的 2D 目标检测模型的启发,我们提出了一种基于点云的 3D 目标检测架构。我们的方法将 3D 目标检测建模为动态图上的消息传递,推广了 DGCNN 框架以预测一组目标。在我们的构建中,我们通过目标置信度聚合或非极大值抑制消除了后处理的必要性。为促进从稀疏点云进行目标检测,我们还提出了一种针对 3D 检测定制的集合到集合蒸馏方法。该方法以置换不变的方式对齐教师模型和学生模型的输出,显著简化了 3D 检测任务的知识蒸馏。我们的方法在自动驾驶基准上取得了最先进的性能。我们还提供了对检测模型和蒸馏框架的丰富分析。

关键词

引用

@article{arxiv.2110.06923,
  title  = {Object DGCNN: 3D Object Detection using Dynamic Graphs},
  author = {Yue Wang and Justin Solomon},
  journal= {arXiv preprint arXiv:2110.06923},
  year   = {2021}
}

备注

Accepted to NeurIPS 2021