改进双流神经网络的遥感图像分类研究

RESEARCH ON REMOTE SENSING IMAGE CLASSIFICATION BASED ON IMPROVED TWO-STREAM NEURAL NETWORK

  • 摘要: 高分辨率遥感图像场景分类是遥感图像处理不可或缺的一部分。针对现有卷积神经网络无法克服遥感图像类内多样性大、类间相似度高的难题,提出一种改进有效通道注意力机制(ECA)结合图卷积神经网络的双流遥感图像场景分类模型——ConvNeXt-SPCECA-GCN。在ECA的基础上引入空间注意力机制,使网络可以从空间和通道维度上提取特征信息;通过图卷积神经网络提取长距离空间信息,采用加法融合策略融合局部关键特征和长距离空间特征实现分类。在实验中采用多种数据增强方法训练模型,有效缓解了数据量不足对模型的影响,提高了模型的泛化能力。最后在UCMerced Land-Use和AID DataSet上进行实验,平均准确率分别达到99.03%和96.87%。

     

    Abstract: High-resolution remote sensing image scene classification is an indispensable part of remote sensing image processing, and existing convolutional neural networks cannot overcome the problems of large intra-class diversity and high inter-class similarity of remote sensing images. This paper proposes a two-stream remote sensing image scene classification model named ConvNeXt-SPCECA-GCN based on improved efficient channel attention (ECA) combined with graph convolutional network. The spatial attention mechanism was introduced on the basis of ECA, so that the network could extract feature information from the spatial and channel dimensions. The long-distance spatial information was extracted through the graph convolutional network, and the additive fusion strategy was used to fuse the local key features and the long-distance space features implement classification. In the experiment, a variety of data augmentation methods were used to train the model, which effectively alleviated the impact of insufficient data on the model and enhanced the generalization ability of the model. Experiments were carried out on the datasets UCMerced Land-Use and AID DataSet, and the average accuracy rates were 99.03% and 96.87%, respectively.

     

/

返回文章
返回