基于双线性注意力机制的遥感场景分类方法

REMOTE SENSING SCENE CLASSIFICATION BASED ON BILINEAR ATTENTION MECHANISM

  • 摘要: 针对遥感影像场景不同类型间相似度高、相同类型内部差异大,导致相似场景容易产生错分的问题,提出一种双线性注意力机制的遥感场景的分类方法。选择轻量化的InceptionV3作为主干网络;引入注意力机制模块,以突出显著的特征信息并忽略无关的背景信息;引入迁移学习提高模型对场景的分类性能。结果表明,该方法分类精度在NWPU45和AID数据集分别达到了97.50%和96.70%,证明所提的方法能够提升遥感图像场景分类精度。

     

    Abstract: Aiming at the problem of high similarity between different types of remote sensing image scenes and large differences within the same type, which leads to the easy misclassification of similar scenes, this paper proposes a classification method of remote sensing scenes with a bilinear attention mechanism. The lightweight InceptionV3 was selected as the backbone network. The attention mechanism module was introduced to highlight salient feature information and ignore irrelevant background information. The transfer learning was introduced to improve the performance of the model for scene classification. The results show that the classification accuracy of our method reaches 97.50% and 96.70% on the NWPU45 and AID datasets, respectively. It is demonstrated that the proposed model can improve the performance of remote sensing images scene classification.

     

/

返回文章
返回