一种基于轻量去冗余卷积和改进Transformer的图像分类方法及系统
申请号:CN202411562742
申请日期:2024-11-05
公开号:CN119418124B
公开日期:2025-12-05
类型:发明专利
摘要
本发明公开了一种基于轻量去冗余卷积和改进Transformer的图像分类方法及系统,属于图像处理技术领域,本发明包括获取分类图像数据集;将分类图像数据集划分为训练集与测试集并进行数据增强;构建轻量去冗余卷积和改进Transformer的图像分类模型并采用迁移学习的训练方式,使用训练集优化参数得到最优模型;加载训练好的最优模型,输入待分类图像得到分类结果。本发明通过构建轻量去冗余卷积对背景相似,目标区域小的图像进行特征筛选,去除冗余特征,获取鲁棒的特征表示,同时采用卷积神经网络和改进的Transformer结合的双流模式加强模型的归纳偏置能力和长距离依赖捕获能力,有效提高了图像分类最终结果的质量,具备适用性强和分类性能高的优良特性。
技术关键词
图像分类方法
分类图像数据
图像分类模型
卷积模块
图像分类系统
训练集优化
采样模块
模型训练模块
交互机制
图像处理技术
冗余特征
数据获取模块
数据处理模块
输出模块
通道
注意力