Journal of Optoelectronics · Laser, Volume. 35, Issue 7, 716(2024)

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

SHU Xinhang1,2,3, WEN Xianbin1,2、*, YUAN Liming1,2, XU Haixia1,2, and SHI Furong1,2
Author Affiliations
  • 1School of Computer Science and Engineering, Tianjin University of Technology, Tianjin 300384, China
  • 2Key Laboratory of Computer Vision and System, Ministry of Education, Tianjin 300384, China
  • 3Center of Research and Development, China Academy of Civil Aviation Science and Technology, Beijing 100028, China
  • show less

    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.

    Tools

    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

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category:

    Received: Nov. 10, 2022

    Accepted: Dec. 13, 2024

    Published Online: Dec. 13, 2024

    The Author Email: WEN Xianbin (xbwen@emial.tjut.edu.cn)

    DOI:10.16136/j.joel.2024.07.0768

    Topics