| 基于迁移学习和通道注意力的遥感图像场景分类 |
| |
| 引用本文： | 舒薪行,温显斌,袁立明,徐海霞,史芙蓉.基于迁移学习和通道注意力的遥感图像场景分类[J].光电子．激光,2024,35(7):716-722. |
| |
| 作者姓名： | 舒薪行 温显斌 袁立明 徐海霞 史芙蓉 |
| |
| 作者单位： | 天津理工大学 计算机科学与工程学院，天津 300384 ;计算机视觉与系统教育部重点实验室，天津 300384 ;中国民航科学技术研究院研发中心,北京 100028,天津理工大学 计算机科学与工程学院，天津 300384 ;计算机视觉与系统教育部重点实验室，天津 300384,天津理工大学 计算机科学与工程学院，天津 300384 ;计算机视觉与系统教育部重点实验室，天津 300384,天津理工大学 计算机科学与工程学院，天津 300384 ;计算机视觉与系统教育部重点实验室，天津 300384,天津理工大学 计算机科学与工程学院，天津 300384 ;计算机视觉与系统教育部重点实验室，天津 300384 |
| |
| 基金项目： | 天津市新一代人工智能科技重大专项基金(18ZXZNGX00150)和天津市技术创新引导专项(21YDTPJC00250)资助项目 |
| |
| 摘 要： | 针对遥感图像场景分类任务中训练样本数量少及遥感图像背景复杂等问题,本文将迁移学习和通道注意力引入到卷积神经网络(convolutional neural network,CNN) 中,提出基于迁移学习和通道注意力的遥感图像场景分类方法。该方法首先选用经过ImageNet自然数据集预训练的两个CNN作为主干,同时引入通道注意力机制,自适应地增强主要特征,抑制次要特征；然后融合这两个网络提取的特征进行分类；最后采用微调迁移学习的方式实现目标域上的学习与分类。提出的方法在几个经典的公共数据集上进行了评估,实验结果证明了本文提出的方法在遥感图像场景分类中达到与其他先进方法相当的性能。
|
| 关 键 词： | 遥感图像 场景分类 卷积神经网络(CNN) 迁移学习 通道注意力 |
| 收稿时间： | 2022/11/10 0:00:00 |
| 修稿时间： | 2023/3/10 0:00:00 |
Remote sensing image scene classification based on transfer learning and channel attention |
| |
| SHU Xinhang,WEN Xianbin,YUAN Liming,XU Haixia and SHI Furong.Remote sensing image scene classification based on transfer learning and channel attention[J].Journal of Optoelectronics·laser,2024,35(7):716-722. |
| |
| Authors: | SHU Xinhang WEN Xianbin YUAN Liming XU Haixia and SHI Furong |
| |
| Affiliation: | School of Computer Science and Engineering, Tianjin University of Technology, Tianjin 300384, China;Key Laboratory of Computer Vision and System, Ministry of Education, Tianjin 300384, China;Center of Research and Development, China Academy of Civil Aviation Science and Technology,Beijing 100028,China,School of Computer Science and Engineering, Tianjin University of Technology, Tianjin 300384, China;Key Laboratory of Computer Vision and System, Ministry of Education, Tianjin 300384, China,School of Computer Science and Engineering, Tianjin University of Technology, Tianjin 300384, China;Key Laboratory of Computer Vision and System, Ministry of Education, Tianjin 300384, China,School of Computer Science and Engineering, Tianjin University of Technology, Tianjin 300384, China;Key Laboratory of Computer Vision and System, Ministry of Education, Tianjin 300384, China and School of Computer Science and Engineering, Tianjin University of Technology, Tianjin 300384, China;Key Laboratory of Computer Vision and System, Ministry of Education, Tianjin 300384, China |
| |
| 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. |
| |
| Keywords: | remote sensing image scene classification convolutional neural network (CNN) transfer learning channel attention |
|
| 点击此处可从《光电子．激光》浏览原始摘要信息 |
|
点击此处可从《光电子．激光》下载全文 |
|
