中文

FocalFormer3D:聚焦困难实例的 3D 目标检测

计算机视觉与模式识别 2023-08-10 v1

摘要

3D 目标检测中的假阴性(FN),例如漏检行人、车辆或其他障碍物,可能在自动驾驶中导致潜在危险情况。尽管致命,该问题在许多当前 3D 检测方法中研究不足。本工作中,我们提出困难实例探测(HIP),一种以多阶段方式识别 \textit{FN} 并引导模型聚焦于挖掘困难实例的通用流程。对于 3D 目标检测,我们将此方法实例化为 FocalFormer3D,一种简单而有效的检测器,擅长挖掘困难目标并提升预测召回率。FocalFormer3D 具有多阶段查询生成以发现困难目标,以及框级 transformer 解码器以高效区分海量候选目标中的物体。在 nuScenes 和 Waymo 数据集上的实验结果验证了 FocalFormer3D 的优越性能。该优势在检测和跟踪、激光雷达与多模态设置下均带来强劲表现。值得注意的是,FocalFormer3D 在 nuScenes 检测基准上达到 70.5 mAP 和 73.9 NDS,而 nuScenes 跟踪基准显示 72.1 AMOTA,两者均排名 nuScenes 激光雷达排行榜第一。我们的代码见于 \url{https://github.com/NVlabs/FocalFormer3D}。

关键词

引用

@article{arxiv.2308.04556,
  title  = {FocalFormer3D : Focusing on Hard Instance for 3D Object Detection},
  author = {Yilun Chen and Zhiding Yu and Yukang Chen and Shiyi Lan and Animashree Anandkumar and Jiaya Jia and Jose Alvarez},
  journal= {arXiv preprint arXiv:2308.04556},
  year   = {2023}
}

备注

Accepted by ICCV 2023