Acta Optica Sinica, Volume. 41, Issue 16, 1611004(2021)

Polarimetric Imaging Target Classification Method Based on Attention Mechanism

Rui Sun1,2,3, Xiaobing Sun1,3、*, Xiao Liu1,3, and Qiang Song1,2,3
Author Affiliations
  • 1Anhui Institute of Optics and Fine Mechanics, Hefei Institutes of Physical Science, Chinese Academy of Sciences, Hefei, Anhui 230031, China
  • 2University of Science and Technology of China, Hefei, Anhui 230026, China
  • 3Key Laboratory of Optical Calibration and Characterization, Chinese Academy of Sciences, Hefei, Anhui 230031, China
  • show less

    The neural network based on attention mechanism can focus on extracting the feature information from the key areas. The application of this characteristic in the polarimetric imaging target classification can help us to obtain the relationships among different polarimetric images and to extract more feature information from critical areas. To solve the difficulty of target recognition in cluttered natural backgrounds, this paper presents a polarimetric imaging target classification method based on an attention mechanism. Firstly, the attention mechanism and the convolutional neural network are combined to construct a polarimetric feature extraction model suitable for limited samples. Then, proper polarimetric images are selected as the input model for training so that the attention module can give more weights to the channel domain and spatial domain feature information that is easily classified to obtain higher classification accuracy. The experimental results show that the classification accuracy of the proposed method can be further improved in different natural backgrounds and reach more than 95% in the self-built polarimetric target database, which is obviously improved to compare with that of the traditional deep learning classification method. Thus, our method is more suitable for target classification in cluttered backgrounds.

    Tools

    Get Citation

    Copy Citation Text

    Rui Sun, Xiaobing Sun, Xiao Liu, Qiang Song. Polarimetric Imaging Target Classification Method Based on Attention Mechanism[J]. Acta Optica Sinica, 2021, 41(16): 1611004

    Download Citation

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

    Category: Imaging Systems

    Received: Jan. 19, 2021

    Accepted: Mar. 18, 2021

    Published Online: Aug. 12, 2021

    The Author Email: Sun Xiaobing (xbsun@aiofm.ac.cn)

    DOI:10.3788/AOS202141.1611004

    Topics