基于改进的MobileViT网络模型的水果分级方法

本发明公开了一种基于改进的MobileViT网络模型的水果分级方法，包括下述步骤：获取水果表面图像数据并进行数据增强和预处理；对预处理后的水果表面图像数据进行图像增强，提高水果表面缺陷区域与正常区域的对比度；将水果表面图像数据划分为训练集、验证集和测试集，标记分级标准；构建水果图像分类网络模型；基于训练集对水果图像分类网络模型进行训练，得到训练后的水果图像分类网络模型；基于测试集对水果图像分类网络模型进行测试，输出水果分级准确率；基于训练后的水果图像分类网络模型输出预测的水果分级结果。本发明可以高效地对采集到的水果图像根据形状、颜色等特征进行分类识别，具有成本较低、适应性好、准确率高等优点。