基于Transformer轻量化模型的图像分类方法

    公开(公告)号:CN116310506A

    公开(公告)日:2023-06-23

    申请号:CN202310090225.X

    申请日:2023-02-09

    Applicant: 燕山大学

    Inventor: 李国强 吴燕芳

    Abstract: 本发明公开了一种基于Transformer轻量化模型的图像分类方法,属于深度学习图像分类的领域,包括以下步骤:S1:对ShuffleNetV2的基本模块和下采样模块进行改进,ShB模块是在基本模块中加入了跳跃连接和ECA,将下采样模块与多特征提取、ECA和Ghost模块相结合提出ShD模块;S2:根据ViT模块提出一种局部和全局信息交互的ShT模块;S3:将ShD与ShB和ShT模块进行有效设计堆叠,构建ShViT网络;S4:ShViT网络在分类数据集上进行训练及测试,得到分类结果。本发明搭建了参数量低、所需内存资源低的轻量化网络模型ShViT,能够在参数量减少的情况下实现较好的分类性能。

Patent Agency Ranking