【鲲码园Python】基于pytorch的蘑菇分类系统(9类)

简介

1. 编程语言: Python;

2. 数据集来自网络mushrooms_9，其中包括6714张蘑菇图片，分9类，依次为伞菌属(Agaricus)、鹅膏属(Amanita)、牛肝菌属(Boletus)、丝膜菌属(Cortinarius)、粉褶蕈属(Entoloma)、蜡伞科(Hygrocybe)、乳菇属(Lactarius)、红菇属(Russula)、乳牛肝菌属(Suillus)；

3. 深度学习使用了ResNet18(残差神经网络，属于CNN深度卷积神经网络的一种)，主要使用pytorch及torchvison库进行模型构建与训练，测试训练轮数为20轮，初始阶段使用了预训练权重文件；

4. 图像转换、预处理使用opencv、torchvision库;

5. GUI界面构建展示主要使用PyQt5库；

6. GUI增加了Grad-CAM热图展示；

内容展示

GUI界面

视频演示：【鲲码园Python】基于pytorch蘑菇分类系统

代码结构

训练结果相关

源码+数据集+模型文件获取

飞书文档：classification_of_mushroom varieties_based_on_pytorch