arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2503.12006cs.CV

ROS-SAM:面向遥感运动目标的高质量交互式分割

ROS-SAM: High-Quality Interactive Segmentation for Remote Sensing Moving Object

Hainan University(海南大学);Zhejiang University(浙江大学);Tianjin University(天津大学)

Zhe Shan, Yang Liu, Lei Zhou, Cheng Yan, Heng Wang, Xia Xie

AI总结:

本文提出 ROS-SAM,通过 LoRA 微调、深层特征增强及全局—局部掩码解码,实现遥感运动目标的高质量交互式分割,并在视频与跟踪数据集上显著提升 IoU、展现零样本能力。

摘要:

大规模遥感视频数据的可得性凸显了高质量交互式分割的重要性。然而,目标尺寸小、特征模糊以及泛化能力有限等挑战,使现有方法难以实现这一目标。在本工作中,我们提出 ROS-SAM,这是一种旨在实现高质量交互式分割、同时保持对多样化遥感数据泛化能力的方法。ROS-SAM 建立在三项关键创新之上:1)基于 LoRA 的微调,可在保持 SAM 泛化能力的同时实现高效领域适配;2)增强深层网络层,以提升所提取特征的判别性,从而减少误分类;3)在掩码解码器中融合全局上下文与局部边界细节,以生成高质量分割掩码。此外,我们设计了数据流水线,确保模型在训练期间学习更好地处理不同尺度的目标,同时在推理时聚焦于高质量预测。在遥感视频数据集上的实验表明,重新设计的数据流水线使 IoU 提升 6%,而 ROS-SAM 使 IoU 提升 13%。最后,在现有遥感目标跟踪数据集上评估时,ROS-SAM 展现出令人瞩目的零样本能力,生成的掩码与人工标注高度相似。这些结果证实 ROS-SAM 是遥感应用中细粒度分割的强大工具。代码可在 https://github.com/ShanZard/ROS-SAM 获取。

英文摘要:

The availability of large-scale remote sensing video data underscores the importance of high-quality interactive segmentation. However, challenges such as small object sizes, ambiguous features, and limited generalization make it difficult for current methods to achieve this goal. In this work, we propose ROS-SAM, a method designed to achieve high-quality interactive segmentation while preserving generalization across diverse remote sensing data. The ROS-SAM is built upon three key innovations: 1) LoRA-based fine-tuning, which enables efficient domain adaptation while maintaining SAM's generalization ability, 2) Enhancement of deep network layers to improve the discriminability of extracted features, thereby reducing misclassifications, and 3) Integration of global context with local boundary details in the mask decoder to generate high-quality segmentation masks. Additionally, we design the data pipeline to ensure the model learns to better handle objects at varying scales during training while focusing on high-quality predictions during inference. Experiments on remote sensing video datasets show that the redesigned data pipeline boosts the IoU by 6%, while ROS-SAM increases the IoU by 13%. Finally, when evaluated on existing remote sensing object tracking datasets, ROS-SAM demonstrates impressive zero-shot capabilities, generating masks that closely resemble manual annotations. These results confirm ROS-SAM as a powerful tool for fine-grained segmentation in remote sensing applications. Code is available at https://github.com/ShanZard/ROS-SAM.

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

↑