中文

Diff3DETR:用于半监督 3D 目标检测的基于代理的扩散模型

计算机视觉与模式识别 2024-08-02 v1

摘要

3D 目标检测是理解 3D 场景的关键技术。当前方法常需大量标注训练数据,但获取点云的点级标注费时费力。近期半监督方法通过教师-学生框架为无标签点云生成伪标签,以缓解此问题。然而,这些伪标签常 suffer from 采样位置和内容嵌入之间的不足多样性和低质量。为克服此困难,我们引入基于代理的扩散模型用于半监督 3D 目标检测(Diff3DETR)。具体而言,设计了基于代理的对象查询生成器,以有效适应动态场景,同时在采样位置和内容嵌入之间取得平衡。此外,box-aware 去噪模块利用 DDIM 去噪过程和变换器解码器中的长程注意力,逐步细化边界框。在 ScanNet 和 SUN RGB-D 数据集上的大量实验表明,Diff3DETR 在半监督 3D 目标检测方法中超越了现有的 SOTA 方法。

关键词

引用

@article{arxiv.2408.00286,
  title  = {Diff3DETR:Agent-based Diffusion Model for Semi-supervised 3D Object Detection},
  author = {Jiacheng Deng and Jiahao Lu and Tianzhu Zhang},
  journal= {arXiv preprint arXiv:2408.00286},
  year   = {2024}
}

备注

Accepted to ECCV2024