Acta Photonica Sinica, Volume. 50, Issue 7, 79(2021)

Remote Sensing Image Scene Classification Based on Supervised Contrastive Learning

Dongen GUO1,2, Ying XIA1, Xiaobo LUO1, and Jiangfan FENG1
Author Affiliations
  • 1Chongqing Engineering Research Center for Spatial Big Data Intelligent Technology, Chongqing University of Posts and Telecommunications, Chongqing400065, China
  • 2School of Computer and Software, Nanyang Institute of Technology, Nanyang,Henan473000, China
  • show less

    To solve the problem of scene classification performance caused by complex background, intra-class diversity and inter-class similarity in remote sensing scene images, a new remote sensing scene classification method based on supervised contrast learning is proposed. The method involves two stages: discriminative feature learning and linear classification. In the stage of discriminative feature learning, a supervised contrastive loss first is introduced to narrow the distance between similar scenes and increase the distance between different types of scenes, so as to improve the scene discriminative ability of intra-class diversity and inter-class similarity; secondly, a gated self-attention module is introduced to filter useless background information and focus on key scene parts for improving scene recognition capabilities with complex backgrounds; finally, the pre-trained Inception V3 branch is introduced, and the branch features are merged with the final features extracted by the original model to further enhance the feature discriminative ability for improving the overall performance of scene classification. In the linear classification stage, the classification results are obtained by fine-tuning the model trained in the first stage. Comprehensive experiments on AID and NWPU-RESISC45 datasets demonstrate the effectiveness of the proposed method.

    Tools

    Get Citation

    Copy Citation Text

    Dongen GUO, Ying XIA, Xiaobo LUO, Jiangfan FENG. Remote Sensing Image Scene Classification Based on Supervised Contrastive Learning[J]. Acta Photonica Sinica, 2021, 50(7): 79

    Download Citation

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

    Category: Image Processing

    Received: Dec. 12, 2020

    Accepted: Mar. 4, 2021

    Published Online: Sep. 1, 2021

    The Author Email:

    DOI:10.3788/gzxb20215007.0710002

    Topics