用语言引导让SAM2高效完成多模态图像分割,提升机器人感知能力。
Unveiling the Potential of Segment Anything Model 2 for RGB-Thermal Semantic Segmentation with Language Guidance
- 基于文本引导的跨模态融合与异构提示解码,打破SAM2对可见光图像的偏好。
- 在PST900和FMB数据集上分别达到89.8%和67.8%的分割精度,性能领先。
- 适合希望低成本部署多模态感知的机器人研究者或工程师使用。
机器人系统的感知能力依赖于数据集的丰富性。尽管基于大规模数据训练的Segment Anything Model 2(SAM2)在感知任务中展现出强大潜力,但其固有的训练范式使其难以适用于RGB-T任务。为解决这一问题,我们提出SHIFNet,一种基于SAM2的新型混合交互范式,通过语言引导实现高效RGB-热成像语义分割。该框架包含两个核心组件:(1) 语义感知跨模态融合(SACF)模块,通过文本引导的亲和力学习动态平衡模态贡献,克服SAM2固有的可见光偏见;(2) 异构提示解码器(HPD),通过语义增强模块强化全局语义信息,并结合类别嵌入以增强跨模态语义一致性。该模型仅含32.27M可训练参数,在公开基准上取得最优表现,分别在PST900和FMB数据集上达到89.8%和67.8%的分割准确率。该框架有效实现了预训练大模型向RGB-T分割任务的迁移,显著降低数据采集成本,赋予机器人系统全面感知能力。源代码将公开于https://github.com/iAsakiT3T/SHIFNet。
原文摘要 · Abstract (English)
The perception capability of robotic systems relies on the richness of the dataset. Although Segment Anything Model 2 (SAM2), trained on large datasets, demonstrates strong perception potential in perception tasks, its inherent training paradigm prevents it from being suitable for RGB-T tasks. To address these challenges, we propose SHIFNet, a novel SAM2-driven Hybrid Interaction Paradigm that unlocks the potential of SAM2 with linguistic guidance for efficient RGB-Thermal perception. Our framework consists of two key components: (1) Semantic-Aware Cross-modal Fusion (SACF) module that dynamically balances modality contributions through text-guided affinity learning, overcoming SAM2's inherent RGB bias; (2) Heterogeneous Prompting Decoder (HPD) that enhances global semantic information through a semantic enhancement module and then combined with category embeddings to amplify cross-modal semantic consistency. With 32.27M trainable parameters, SHIFNet achieves state-of-the-art segmentation performance on public benchmarks, reaching 89.8% on PST900 and 67.8% on FMB, respectively. The framework facilitates the adaptation of pre-trained large models to RGB-T segmentation tasks, effectively mitigating the high costs associated with data collection while endowing robotic systems with comprehensive perception capabilities. The source code will be made publicly available at https://github.com/iAsakiT3T/SHIFNet.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。