1. 使用迁移学习+集成学习的思想来实现对于常见水果的好坏进行分类,使用Pytorch官方基于imageNet所训练好成熟框架体系,选择了机器学习界十分流行的四大网络架构densenet, googlenet,resnet,efficientnet使用迁移学习的办法,只修改其输出层的相关参数,进行冻结训练,在迭代训练过10次之后,对模型进行了解冻训练,使其可以修改整个网络结构参数,让我们的网络结构更加的贴合我们实际所使用的分类图像,同时四个网络训练框架训练完成之后,采用集成学习的思想,将四个框架模型进行了集成得到了最终的网络架构
2. 同时为了提升数据的泛化性,对实验数据进行了随机的裁剪,色调调节,归一化标准化,随机放射变化等一系列图像的预处理和图像增广
3. 使用了PyQt工具包进行了应用拓展,可以使用网上自由选择一张图片来进行水果的识别.
4.训练过程中使用了tesorboard对训练集和验证集的参数进行记录,保存已经训练好的模型参数,最后测试模型时绘制多分类的ROC曲线,与混淆矩阵
5.文件有相关的代码,预训练好的模型,相关的文档,还有ppt,数据集,和使用说明