arXiv:2410.16545cs.CV2024-10被引 4

融合深度与彩色信息,提升平面实例分割精度

PlaneSAM: Multimodal Plane Instance Segmentation Using the Segment Anything Model

  • 双复杂度主干网络分别学习深度与彩色特征,兼顾效率与性能
  • 在ScanNet上达到新SOTA,零样本迁移表现优于现有方法
  • 仅增加10%计算开销,支持全自动平面分割,适合实际部署

基于RGB-D数据的平面实例分割对众多下游任务至关重要。现有深度学习方法大多仅利用彩色图像信息,忽略了深度图在平面分割中的重要作用。本文提出PlaneSAM,一种基于EfficientSAM的快速版本,通过双复杂度主干网络,分别高效学习深度(D-band)和彩色(RGB-band)特征。该设计在深度数据有限时仍能有效提取几何特征,同时保留EfficientSAM强大的彩色特征表示能力,并支持原骨干网络微调。为增强对RGB-D域的适应性,采用基于不完美伪标签的自监督预训练策略,在大规模RGB-D数据上进行分割任意任务预训练。针对大平面分割需求,优化了EfficientSAM的损失函数组合比例。此外,使用Faster R-CNN作为平面检测器,其预测框作为提示输入双复杂度网络,实现完全自动化的平面实例分割。实验表明,PlaneSAM在ScanNet数据集上达到新SOTA,且在2D-3D-S、Matterport3D和ICL-NUIM等数据集的零样本迁移中表现更优,计算开销仅比EfficientSAM增加10%。

原文摘要 · Abstract (English)

Plane instance segmentation from RGB-D data is a crucial research topic for many downstream tasks. However, most existing deep-learning-based methods utilize only information within the RGB bands, neglecting the important role of the depth band in plane instance segmentation. Based on EfficientSAM, a fast version of SAM, we propose a plane instance segmentation network called PlaneSAM, which can fully integrate the information of the RGB bands (spectral bands) and the D band (geometric band), thereby improving the effectiveness of plane instance segmentation in a multimodal manner. Specifically, we use a dual-complexity backbone, with primarily the simpler branch learning D-band features and primarily the more complex branch learning RGB-band features. Consequently, the backbone can effectively learn D-band feature representations even when D-band training data is limited in scale, retain the powerful RGB-band feature representations of EfficientSAM, and allow the original backbone branch to be fine-tuned for the current task. To enhance the adaptability of our PlaneSAM to the RGB-D domain, we pretrain our dual-complexity backbone using the segment anything task on large-scale RGB-D data through a self-supervised pretraining strategy based on imperfect pseudo-labels. To support the segmentation of large planes, we optimize the loss function combination ratio of EfficientSAM. In addition, Faster R-CNN is used as a plane detector, and its predicted bounding boxes are fed into our dual-complexity network as prompts, thereby enabling fully automatic plane instance segmentation. Experimental results show that the proposed PlaneSAM sets a new SOTA performance on the ScanNet dataset, and outperforms previous SOTA approaches in zero-shot transfer on the 2D-3D-S, Matterport3D, and ICL-NUIM RGB-D datasets, while only incurring a 10% increase in computational overhead compared to EfficientSAM.

实例分割多模态深度估计SAM

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。