arXiv:2409.13401cs.CV2024-09中稿 · IEEE TGRS被引 93

用点标注训练遥感图像分割模型,效果优于传统方法。

PointSAM: Pointly-Supervised Segment Anything Model for Remote Sensing Images

  • 用点标注替代密集掩码,通过自训练生成伪标签。
  • 在多个遥感数据集上显著超越SAM、SAM2等基线模型。
  • 提出负提示校准,解决重叠目标误分割问题,适合遥感领域。

Segment Anything Model(SAM)是先进的图像分割基础模型,正逐步应用于遥感图像(RSI)。由于遥感图像与自然图像存在领域差异,传统方法通常将SAM作为预训练模型,并使用全监督掩码进行微调。本文提出一种新方法,仅使用更便捷且更具挑战性的点标注进行微调。利用SAM的零样本能力,采用自训练框架迭代生成伪标签。为避免伪标签噪声导致误差累积,从目标数据集中提取目标原型,并通过匈牙利算法匹配预测原型,防止模型学习错误方向。此外,针对遥感图像背景复杂、目标密集的问题,单点提示可能导致多个对象被识别为一个,为此提出基于实例掩码非重叠特性的负提示校准方法:将重叠掩码的提示作为负信号,实现掩码精细化。综合上述方法,提出名为PointSAM的新型点标注监督分割模型。在WHU、HRSID和NWPU VHR-10等遥感数据集上的实验表明,该方法显著优于直接测试SAM、SAM2及其他对比方法。进一步将PointSAM用于点到框转换任务,也取得良好结果,表明其可拓展至其他点监督任务。代码已开源。

原文摘要 · Abstract (English)

Segment Anything Model (SAM) is an advanced foundational model for image segmentation, which is gradually being applied to remote sensing images (RSIs). Due to the domain gap between RSIs and natural images, traditional methods typically use SAM as a source pre-trained model and fine-tune it with fully supervised masks. Unlike these methods, our work focuses on fine-tuning SAM using more convenient and challenging point annotations. Leveraging SAM's zero-shot capabilities, we adopt a self-training framework that iteratively generates pseudo-labels for training. However, if the pseudo-labels contain noisy labels, there is a risk of error accumulation. To address this issue, we extract target prototypes from the target dataset and use the Hungarian algorithm to match them with prediction prototypes, preventing the model from learning in the wrong direction. Additionally, due to the complex backgrounds and dense distribution of objects in RSI, using point prompts may result in multiple objects being recognized as one. To solve this problem, we propose a negative prompt calibration method based on the non-overlapping nature of instance masks. In brief, we use the prompts of overlapping masks as corresponding negative signals, resulting in refined masks. Combining the above methods, we propose a novel Pointly-supervised Segment Anything Model named PointSAM. We conduct experiments on RSI datasets, including WHU, HRSID, and NWPU VHR-10, and the results show that our method significantly outperforms direct testing with SAM, SAM2, and other comparison methods. Furthermore, we introduce PointSAM as a point-to-box converter and achieve encouraging results, suggesting that this method can be extended to other point-supervised tasks. The code is available at https://github.com/Lans1ng/PointSAM.

遥感分割点标注自训练SAM

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