基于改进的MobileViT模型的图像分类方法、电子设备及可读存储介质

AITNT
正文
推荐专利
基于改进的MobileViT模型的图像分类方法、电子设备及可读存储介质
申请号:CN202410953771
申请日期:2024-07-16
公开号:CN118840608A
公开日期:2024-10-25
类型:发明专利
摘要
本发明提供的一种基于改进的MobileViT模型的图像分类方法,包括以下步骤:S1.获取图像数据集,并对图像数据集进行预处理,得到样本数据集;S2.构建改进的MobileViT模型;S3.将样本数据集输入至改进的MobileViT模型中进行训练;S4.判断改进的MobileViT模型是否训练完成,如是,则进入步骤S5;如否,则更新参数,并返回步骤S3中;S5.对待分类图像进行预处理,并将预处理后的图像输入至训练完成的改进的MobileViT模型中,得到分类结果。
技术关键词
图像分类方法 卷积模块 注意力机制 轻量级神经网络 全局特征提取 局部特征提取 图像块 样本 输入端 可读存储介质 存储计算机程序 数据 输出端 电子设备 处理器 亮度 存储器
系统为您推荐了相关专利信息
1
基于多参数同步测量的智能电表故障预警方法及系统
智能电表故障 图像特征向量 预警方法 交叉注意力机制 多参数
2
一种排班排产模型设计方法、装置、存储介质和程序产品
启发式算法 智能网络 排产模型 数据抽样 多头注意力机制
3
一种肌电信号识别分类方法及系统
肌电信号识别 频域特征 分类方法 校准 多元线性回归算法
4
基于时空图卷积网络的多源数据交易风险预测方法及系统
交易风险预测 风险预测模型 融合特征 深度特征融合 多视角
5
一种基于Transformer自适应对抗图神经网络的交通流预测方法
交通流预测方法 注意力机制 随机梯度下降 交通流预测系统 数据
添加客服微信openai178,进AITNT官方交流群
驱动智慧未来:提供一站式AI转型解决方案
沪ICP备2023015588号