基于迁移学习和通道注意力的遥感图像场景分类
CSTR:
作者:
作者单位:

(1.天津理工大学 计算机科学与工程学院,天津 300384; 2.计算机视觉与系统教育部重点实验室,天津 300384; 3.中国民航科学技术研究院研发中心,北京 100028)

作者简介:

温显斌 (1966-),男,博士,教授,博士生导师,主要从事计算机视觉、遥感图像理解、多媒体分析等方面的研究。

通讯作者:

中图分类号:

基金项目:

天津市新一代人工智能科技重大专项基金(18ZXZNGX00150)和天津市技术创新引导专项(21YDTPJC00250)资助项目


Remote sensing image scene classification based on transfer learning and channel attention
Author:
Affiliation:

(1.School of Computer Science and Engineering, Tianjin University of Technology, Tianjin 300384, China;2.Key Laboratory of Computer Vision and System, Ministry of Education, Tianjin 300384, China;3.Center of Research and Development, China Academy of Civil Aviation Science and Technology,Beijing 100028,China)

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对遥感图像场景分类任务中训练样本数量少及遥感图像背景复杂等问题,本文将迁移学习和通道注意力引入到卷积神经网络(convolutional neural network,CNN) 中,提出基于迁移学习和通道注意力的遥感图像场景分类方法。该方法首先选用经过ImageNet自然数据集预训练的两个CNN作为主干,同时引入通道注意力机制,自适应地增强主要特征,抑制次要特征;然后融合这两个网络提取的特征进行分类;最后采用微调迁移学习的方式实现目标域上的学习与分类。提出的方法在几个经典的公共数据集上进行了评估,实验结果证明了本文提出的方法在遥感图像场景分类中达到与其他先进方法相当的性能。

    Abstract:

    In order to solve the problems of small number of training samples and complex background of remote sensing images,this paper introduces transfer learning and channel attention into convolutional neural network (CNN),and proposes a remote sensing image scene classification method based on transfer learning and channel attention.Firstly,this method selects two CNNs pre-trained by ImageNet natural dataset as the backbone,and introduces the channel attention mechanism to adaptively enhance the main features and suppress the secondary features.Then the features extracted from these two networks are fused for classification.Finally,fine-tuning transfer learning is used to realize learning and classification in the target domain.The proposed method is evaluated on several classical public datasets,and the experimental results show that the proposed method achieves the same performance as other advanced methods in remote sensing image scene classification.

    参考文献
    相似文献
    引证文献
引用本文

舒薪行,温显斌,袁立明,徐海霞,史芙蓉.基于迁移学习和通道注意力的遥感图像场景分类[J].光电子激光,2024,35(7):716~722

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2022-11-10
  • 最后修改日期:2023-03-10
  • 录用日期:
  • 在线发布日期: 2024-06-05
  • 出版日期:
文章二维码