基于SF-Unet的高分辨率耕地遥感影像分割  

Segmentation of High Resolution Remote Sensing Images of Cultivated Land Based on SF-Unet

作　　者：秦伦明 凌雪海 邹钰洁 龚杨洋 边后琴 王悉[3] QIN Lunming;LING Xuehai;ZOU Yujie;GONG Yangyang;BIAN Houqin;WANG Xi(College of Electronics and Information Engineering,Shanghai University of Electric Power,Shanghai 201306,China;Shanghai Zhabei Power Plant of State Grid Corporation of China,Shanghai 200432,China;School of Electronic and Information Engineering,Beijing Jiaotong University,Beijing 100044,China)

机构地区：[1]上海电力大学电子信息工程学院,上海201306 [2]国网上海市电力公司闸北发电厂,上海200432 [3]北京交通大学电子信息工程学院,北京100044

出　　处：《无线电工程》2024年第5期1197-1204,共8页Radio Engineering

基　　金：国家自然科学基金面上项目(62073024)。

摘　　要：利用深度学习实现遥感影像耕地区域自动化检测,取代人工解译,能有效提升耕地面积统计效率。针对目前存在分割目标尺度大且连续导致分割区域存在欠分割现象,边界区域情况复杂导致边缘分割困难等问题,提出了语义分割算法——Swin Transformer,TransFuse and U-Net(SF-Unet)。为强化网络不同层次特征提取和信息融合能力,提升边缘分割性能,使用U-Net网络替代TransFuse网络中的ResNet50模块;将Vision Transformer(ViT)替换为改进后的Swin Transformer网络,解决大区域的欠分割问题;通过注意力机制构建的Fusion融合模块将2个网络输出特征进行融合,增强模型对目标的语义表示,提高分割的精度。实验表明,SF-Unet语义分割网络在Gaofen Image Dataset(GID)数据集上的交并比(Intersection over Union,IoU)达到了90.57%,分别比U-Net和TransFuse网络提升了6.48%和6.09%,明显提升了耕地遥感影像分割的准确性。Using deep learning to realize automatic detection of cultivated land areas in remote sensing images,replacing manual interpretation,can effectively improve the statistical efficiency of cultivated land areas.Aiming at the current problems that the segmentation target scale is large and continuous,which leads to under-segmentation in the segmented area,and the complexity of the boundary area makes edge segmentation difficult,the Swin Transformer,TransFuse and U-Net(SF-Unet)semantic segmentation algorithm is proposed.In order to strengthen the feature extraction and information fusion capabilities of different levels of the network and improve the edge segmentation performance,the U-Net network is used to replace the ResNet50 module in the TransFuse network;the Vision Transformer(ViT)is replaced by improved Swin Transformer network to solve the problem of under-segmentation of large areas;the Fusion module constructed by the attention mechanism fuses the output features of the two networks to enhance the semantic representation of the model to the target and improve the accuracy of segmentation.Experiments show that the Intersection over Union(IoU)ratio of the SF-Unet semantic segmentation network on the Gaofen Image Dataset(GID)has reached 90.57%,which is 6.48%and 6.09%higher than U-Net and TransFuse networks,respectively.The accuracy of cultivated land remote sensing image segmentation is significantly improved.

关 键 词：耕地遥感影像 语义分割 卷积神经网络 Swin Transformer 

分 类 号：TP391[自动化与计算机技术—计算机应用技术]