3 个月前

基于投票的点云3D物体检测中的回溯代表性点

基于投票的点云3D物体检测中的回溯代表性点

摘要

基于点云的3D物体检测是一项具有挑战性的视觉任务,广泛应用于理解三维视觉世界。近年来,大量研究聚焦于如何利用端到端可训练的霍夫投票(Hough voting)机制生成物体候选区域。然而,现有的投票策略仅能从潜在物体表面接收部分有效投票,同时受到杂乱背景中大量离群点的干扰,导致输入点云中的信息未能被充分挖掘和利用。受传统霍夫投票方法中回溯(back-tracing)策略的启发,本文提出一种新型3D物体检测方法——回溯代表性点网络(Back-tracing Representative Points Network, BRNet)。该方法通过生成式地从投票中心回溯代表性点,并进一步重新审视这些生成点周围的互补种子点,从而更有效地捕捉原始点云中潜在物体周围的精细局部结构特征。由此,BRNet采用自底向上再自顶向下的协同策略,强化了预测投票中心与原始表面点之间的相互一致性,显著提升了物体定位与类别预测的可靠性与灵活性。所提出的BRNet结构简洁而高效,在两个大规模点云数据集上均显著超越当前最先进方法:在ScanNet V2数据集上,mAP@0.50提升7.5%;在SUN RGB-D数据集上,mAP@0.50提升4.7%。同时,该模型保持轻量化与高效率。代码将开源,地址为:https://github.com/cheng052/BRNet。

代码仓库

cheng052/BRNet
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
3d-object-detection-on-scannetv2BRNet
mAP@0.25: 66.1
mAP@0.5: 50.9
3d-object-detection-on-sun-rgbd-valBRNet(Geo only)
mAP@0.25: 61.1
mAP@0.5: 43.7

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于投票的点云3D物体检测中的回溯代表性点 | 论文 | HyperAI超神经