
基于经典轻量级CNN网络ShuffleNet的图像分类项目:菠萝成熟度的八阶段识别
5星
- 浏览量: 0
- 大小:None
- 文件类型:7Z
简介:
本项目采用经典的轻量级CNN网络ShuffleNet,专注于开发菠萝成熟度的八阶段自动识别系统,以提高水果分级效率和准确性。
【基于ShuffleNet网络对8种不同阶段的菠萝成熟度分类】
项目包含代码、数据集和训练好的权重文件,可以直接运行。
项目总大小:201MB
本数据集分为8类:没熟、半熟、成熟等等。
下载解压后的图像目录包括:
- 训练集(4808张图片)
- 测试集(806张图片)
data-train 文件夹中包含训练集,每个子文件夹存放同类别的图像,并以分类类别命名;
data-test 文件夹中包含测试集,同样地,每个子文件夹放有同一类别的图像。
【项目介绍】ShuffleNet的参数量约为一百万左右。网络在训练过程中采用了cos 学习率自动衰减策略,在经过50个epoch的训练后,模型在测试集中达到了87%的最佳精度表现;增加更多的epochs可以进一步提高精度。
run_results 目录下存放了最佳权重文件、以及训练日志和loss与精度曲线等信息。
预测时只需运行predict即可。代码会自动对inference目录下的所有图片进行推理,并将前三个概率最大的类别绘制在左上角显示出来。
【训练自己的数据】请参考readme文档,无需做任何修改,因为代码可以自动生成所需的参数设置(例如分类类别的数量等)。
全部评论 (0)
还没有任何评论哟~


