首页|遥感图像农田识别的跨类别小样本分割方法

遥感图像农田识别的跨类别小样本分割方法

扫码查看
深度学习驱动的遥感图像语义分割方法依赖大量人工标记样本,且对未知任务泛化能力差,尤其是在类别体系不断更新的精细化语义分割任务中,未知类别(训练样本中不存在的类别)的识别精度亟待提升.基于此,本文面向多类农田类别,提出了一种跨类别小样本分割方法.该方法设计了一种双分支结构,即支持分支与查询分支,其中支持分支用于分割先验的提取,查询分支用于完成分割先验的传播,获取查询图像的分割结果.此外,该方法应用查询特征生成自支持查询原型,显著提高了原型的表达能力;在支持集和查询集之间引入原型对齐正则化机制,充分利用来自支持集的知识,提升了分割器的判别能力.试验同时引入高空间分辨率和高光谱图像土地覆盖数据集,对本文方法性能进行了充分验证.试验结果表明,相较于现有小样本分割方法,本文方法在小样本条件下能够获得更为优异的跨类别农田识别结果.
Cross-category few-shot segmentation for farmland recognition in remote sensing images
Deep learning-driven semantic segmentation methods for remote sensing images rely heavily on a large number of manually labeled samples and exhibit poor generalization for unknown tasks,especially in the fine-grained semantic segmentation task where the category system is constantly updated,and the recognition accuracy of the unknown categories(the categories that don't exist in the training samples)needs to be urgently improved.Based on this,the paper proposes a cross-category few-shot segmentation method aimed at multiple farmland categories.The method designs a dual-branch structure,comprising a support branch and a query branch,where the support branch is used for the extraction of segmentation prior,and the query branch is used to complete the propagation of segmentation prior and obtain the segmentation results of the query image.Additionally,the method applies query features to generate self-supporting query prototypes,which significantly improves the expressive ability of the prototypes;a regularization mechanism for prototype alignment between the support and query set is introduced,which makes full use of the knowledge from the support set and improves the discriminative ability of the segmentation.The experiments simultaneously introduce high spatial resolution and hyperspectral image land cover datasets to fully validate the performance of the proposed method.The experimental results show that compared with the existing few-shot segmentation methods,the proposed method can obtain more excellent cross-category farmland recognition results under few-shot conditions.

remote sensing imagessemantic segmentationfew-shot learningprototype learning

王星、倪欢

展开 >

南京信息工程大学遥感与测绘工程学院,江苏 南京 210044

遥感图像 语义分割 小样本学习 原型学习

先进光学遥感技术北京市重点实验室开放基金

AORS202310

2024

测绘通报
测绘出版社

测绘通报

CSTPCD北大核心
影响因子:1.027
ISSN:0494-0911
年,卷(期):2024.(10)