arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2312.02464cs.CV

SAM辅助的遥感图像语义分割与对象和边界约束

SAM-Assisted Remote Sensing Imagery Semantic Segmentation with Object and Boundary Constraints

The Chinese University of Hong Kong, Shenzhen(香港中文大学(深圳));Wuhan University of Science and Technology(武汉科技大学);The Hong Kong Polytechnic University(香港理工大学);The University of Hong Kong(香港大学)

Xianping Ma, Qianqian Wu, Xingyu Zhao, Xiaokang Zhang, Man-On Pun, Bo Huang

AI总结:

本文提出一种利用SAM生成对象和边界约束的遥感图像语义分割框架,通过对象损失和边界损失提升分割性能,并在ISPRS Vaihingen和LoveDA Urban数据集上验证了有效性。

摘要:

遥感图像的语义分割在提取各种下游应用的精确信息中起着关键作用。近期,通用分割模型Segment Anything Model(SAM)的发展彻底改变了这一领域,为准确高效的分割提供了新途径。然而,SAM仅限于生成没有类别信息的分割结果。因此,如何利用如此强大的通用视觉模型进行遥感图像的语义分割已成为研究焦点。在本文中,我们提出了一种精简框架,旨在利用SAM的原始输出,通过引入两个新概念:SAM生成对象(SGO)和SAM生成边界(SGB)。具体而言,我们提出了一种新颖的对象损失,并进一步引入边界损失作为增强组件,以在通用语义分割框架中辅助模型优化。考虑到SGO的内容特征,我们引入了对象一致性的概念,以利用缺乏语义信息的分割区域。通过施加对象内预测值一致性的约束,对象损失旨在提升语义分割性能。此外,边界损失利用SGB的独特特征,将模型的注意力引导到对象的边界信息上。在两个知名数据集(即ISPRS Vaihingen和LoveDA Urban)上的实验结果表明了我们提出方法的有效性。本工作的源代码将在https://github.com/sstary/SSRS上提供。

英文摘要:

Semantic segmentation of remote sensing imagery plays a pivotal role in extracting precise information for diverse down-stream applications. Recent development of the Segment Anything Model (SAM), an advanced general-purpose segmentation model, has revolutionized this field, presenting new avenues for accurate and efficient segmentation. However, SAM is limited to generating segmentation results without class information. Consequently, the utilization of such a powerful general vision model for semantic segmentation in remote sensing images has become a focal point of research. In this paper, we present a streamlined framework aimed at leveraging the raw output of SAM by exploiting two novel concepts called SAM-Generated Object (SGO) and SAM-Generated Boundary (SGB). More specifically, we propose a novel object loss and further introduce a boundary loss as augmentative components to aid in model optimization in a general semantic segmentation framework. Taking into account the content characteristics of SGO, we introduce the concept of object consistency to leverage segmented regions lacking semantic information. By imposing constraints on the consistency of predicted values within objects, the object loss aims to enhance semantic segmentation performance. Furthermore, the boundary loss capitalizes on the distinctive features of SGB by directing the model's attention to the boundary information of the object. Experimental results on two well-known datasets, namely ISPRS Vaihingen and LoveDA Urban, demonstrate the effectiveness of our proposed method. The source code for this work will be accessible at https://github.com/sstary/SSRS.

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

↑