结合注意力与特征融合的遥感建筑物提取方法

A BUILDING EXTRACTION METHOD FOR REMOTE SENSING WITH ATTENTION AND FEATURE FUSION

  • 摘要: 深度学习是遥感影像建筑物提取的重要技术之一。针对目前卷积神经网络在提取建筑物时存在边缘模糊、不同尺寸建筑物提取结果差异大、模型参数量大等问题,提出一种基于注意力和多尺度特征融合的提取方法。利用高效通道注意力模块增强重要特征在网络训练的作用;嵌入多尺度特征深度融合模块提取并交互融合特征,同时借助卷积通道剪枝思想压缩模型。实验表明,该方法具有优异的建筑物提取能力,网络细节感知能力更强,提取边缘更加清晰,对复杂场景下不同尺寸和不规则建筑物的提取效果更好,并很好地平衡了模型提取精度和运行效率。

     

    Abstract: Deep learning is one of the important technologies for building extraction from remote sensing images. Aimed at the problems that the current convolutional neural network method has blurred edges, large differences in the extraction results of buildings of different sizes, and large amount of model parameters when extracting buildings, a new method based on attention and multi-scale feature fusion is proposed. The efficient channel attention module was used to enhance the role of important features in network training. The multi-scale feature deep fusion module was embedded to extract and interactively fuse features, and at the same time, the convolution channel pruning idea was used to compress the model. Experiments show that the method have excellent extraction ability, the detail-awareness of the network is stronger, the extracted edges are clearer, the extraction effect is better for buildings with different sizes and irregularities in complex scenes, and the model extraction accuracy and operation efficiency are well balanced.

     

/

返回文章
返回