-
公开(公告)号:CN118840608A
公开(公告)日:2024-10-25
申请号:CN202410953771.6
申请日:2024-07-16
Applicant: 重庆交通大学
IPC: G06V10/764 , G06V10/82 , G06V10/42 , G06V10/44 , G16H50/20 , G06N3/0464 , G06N3/08
Abstract: 本发明提供的一种基于改进的MobileViT模型的图像分类方法,包括以下步骤:S1.获取图像数据集,并对图像数据集进行预处理,得到样本数据集;S2.构建改进的MobileViT模型;S3.将样本数据集输入至改进的MobileViT模型中进行训练;S4.判断改进的MobileViT模型是否训练完成,如是,则进入步骤S5;如否,则更新参数,并返回步骤S3中;S5.对待分类图像进行预处理,并将预处理后的图像输入至训练完成的改进的MobileViT模型中,得到分类结果。