arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2107.04013cs.CVcs.AIcs.LGcs.RO
收藏

三维目标检测的多模态任务级联

Multi-Modality Task Cascade for 3D Object Detection

Carnegie Mellon University(卡内基梅隆大学)

Jinhyung Park, Xinshuo Weng, Yunze Man, Kris Kitani

摘要:

点云和RGB图像是3D视觉理解的天然互补模式-前者提供对象上稀疏但准确的点位置,而后者包含密集的颜色和纹理信息。尽管有这种紧密传感器融合的潜力,但许多方法训练两个模型,并使用简单的特征级联来表示3D传感器数据。这种分离的训练方案导致潜在的次优性能,并阻止3D任务被用于使2D任务受益,而2D任务本身通常是有用的。为了提供更集成的方法,我们提出了一种新的多模态任务级联网络(MTC-RCNN),该网络利用3D框建议来改进2D分割预测,然后用于进一步细化3D框。我们发现,包括两个阶段的3D模块之间的2D网络显着提高2D和3D任务的性能。此外,为了防止3D模块过度依赖过拟合的2D预测,我们提出了一种双头2D分割训练和推理方案,允许第二个3D模块学习解释不完美的2D分割预测。在具有挑战性的SUN RGB-D数据集上评估我们的模型,我们在单模态和融合网络的最新结果基础上进行了大幅改进($\textbf{+3.8}$ mAP@0.5)。代码将发布$\href{https://github.com/Divadi/MTC_RCNN}{\text{here.}}$

英文摘要:

Point clouds and RGB images are naturally complementary modalities for 3D visual understanding - the former provides sparse but accurate locations of points on objects, while the latter contains dense color and texture information. Despite this potential for close sensor fusion, many methods train two models in isolation and use simple feature concatenation to represent 3D sensor data. This separated training scheme results in potentially sub-optimal performance and prevents 3D tasks from being used to benefit 2D tasks that are often useful on their own. To provide a more integrated approach, we propose a novel Multi-Modality Task Cascade network (MTC-RCNN) that leverages 3D box proposals to improve 2D segmentation predictions, which are then used to further refine the 3D boxes. We show that including a 2D network between two stages of 3D modules significantly improves both 2D and 3D task performance. Moreover, to prevent the 3D module from over-relying on the overfitted 2D predictions, we propose a dual-head 2D segmentation training and inference scheme, allowing the 2nd 3D module to learn to interpret imperfect 2D segmentation predictions. Evaluating our model on the challenging SUN RGB-D dataset, we improve upon state-of-the-art results of both single modality and fusion networks by a large margin ($\textbf{+3.8}$ mAP@0.5). Code will be released $\href{https://github.com/Divadi/MTC_RCNN}{\text{here.}}$

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑