Journal of Optoelectronics · Laser, Volume. 35, Issue 7, 716(2024)
Remote sensing image scene classification based on transfer learning and channel attention
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.
Get Citation
Copy Citation Text
SHU Xinhang, WEN Xianbin, YUAN Liming, XU Haixia, SHI Furong. Remote sensing image scene classification based on transfer learning and channel attention[J]. Journal of Optoelectronics · Laser, 2024, 35(7): 716
Category:
Received: Nov. 10, 2022
Accepted: Dec. 13, 2024
Published Online: Dec. 13, 2024
The Author Email: WEN Xianbin (xbwen@emial.tjut.edu.cn)