吉林大学学报(理学版) ›› 2025, Vol. 63 ›› Issue (3): 795-0803.

• • 上一篇    下一篇

基于空间金字塔注意力的细粒度图像分类

朱丽1, 潘鑫1, 付海涛1, 杨亚杰1, 金晨磊2, 冯宇轩1, 范健3,4   

  1. 1. 吉林农业大学 信息技术学院, 长春 130118; 2. 广东理工学院 信息技术学院, 广东 肇庆 526000;
    3. 长春理工大学 电子信息工程学院, 长春 130022; 4. 吉林省互联网技术支撑中心, 长春 130061
  • 收稿日期:2023-12-29 出版日期:2025-05-26 发布日期:2025-05-26
  • 通讯作者: 冯宇轩 E-mail:fengyuxuan.cn@163.com

Fine-Grained Image Classification Based on Spatial Pyramid Attention

ZHU Li1, PAN Xin1, FU Haitao1, YANG Yajie1, JIN Chenlei2, FENG Yuxuan1, FAN Jian3,4   

  1. 1. College of Information Technology, Jilin Agricultural University, Changchun 130118, China;
    2. School of Information Technology, Guangdong Technology College,  Zhaoqing 526000, Guangdong Province, China;
    3. School of Electronic and Information Engineering, Changchun University of Science and Technology, Changchun 130022, China; 
    4. Jilin Province Internet Technology Support Center, Changchun 130061, China
  • Received:2023-12-29 Online:2025-05-26 Published:2025-05-26

摘要: 基于改进的空间金字塔注意力模块, 提升轻量化网络在细粒度图像分类任务上的性能. 通过结合全局特征与局部特征, 改进模型在不显著增加参数量的情况下提高了轻量化网络的分类性能. 在斯坦福狗数据集上的实验结果表明, 采用该模块的轻量化网络准确率显著提升, 甚至超过部分经典模型. 该方法拓展了轻量化网络在资源受限设备上的应用范围, 为细粒度图像分类问题提供了一种高效、 低计算成本的解决方案.

关键词: 图像分类, 细粒度分类, 注意力机制, 轻量化网络

Abstract: Based on an improved  spatial pyramid attention module, we  enhanced the performance of lightweight networks in fine-grained image classification tasks. By combining global and local features, the  improved model enhanced the classification performance of lightweight networks without significantly increasing the number of parameters. The experimental results  on the Stanford Dogs dataset show  that the lightweight network equipped with this module significantly improves accuracy, even surpassing some classical models. This method expands the application scope of lightweight networks on resource-constrained devices and provides an efficient and low-computational-cost solution for fine-grained image classification problems.

Key words:  , image classification, fine-grained classification, attention mechanism, lightweight network

中图分类号: 

  • TP391.41