Optical Instruments, Volume. 46, Issue 6, 55(2024)

Fine-grained image classification algorithm combining saliency and non-local module

Chen LING, Rongfu ZHANG*, Ziye YANG, Guyu GAO, and Fuqiang ZHAO
Author Affiliations
  • School of Optical-Electrical and Computer Engineering, University of Shanghai for Science and Technology, Shanghai 200093, China
  • show less

    Aiming at the problems of inaccurate discriminative feature acquisition and insufficient use of training data in fine-grained image classification tasks, a fine-grained image classification algorithm combining significance and non-local modules is proposed. By clipping the significance region of four training images and stitching them into one training image, the training data could be enriched and enhanced. Moreover, the non-local module was embedded into the bottleneck module in the high-dimensional feature layer of the ResNet-50 model to connect the four salience regions of the enhanced image, which strengthened the model's attention to the global context information. On the Stanford Cars and CUB-200-2011 data sets, the accuracy of Top-1 classification was 94.01% and 85.97%, respectively. This method performs better than compared data augmentation methods and fine-grained image classification algorithms.

    Keywords
    Tools

    Get Citation

    Copy Citation Text

    Chen LING, Rongfu ZHANG, Ziye YANG, Guyu GAO, Fuqiang ZHAO. Fine-grained image classification algorithm combining saliency and non-local module[J]. Optical Instruments, 2024, 46(6): 55

    Download Citation

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

    Category:

    Received: Dec. 5, 2023

    Accepted: --

    Published Online: Jan. 21, 2025

    The Author Email: ZHANG Rongfu (zrf@usst.edu.cn)

    DOI:10.3969/j.issn.1005-5630.202312050131

    Topics