茶叶病害识别(Python代码,pyTorch框架,深度卷积网络模型,很容易替换为其它模型,带有GUI识别界面)

代码运行要求:Torch库>=1.13.1即可

1.茶叶病害数据集(7类病害和1种正常)

 1.茶叶病害数据集介绍(这个茶病数据集包含茶叶,显示了茶的7种常见疾病:

红叶斑

藻类叶斑

bird eye spot;

灰枯病;

白点;

炭疽病;

褐色枯萎病。

茶叶病害数据集还包含一类健康茶叶。每个类都包含100多个图像)

茶叶病害识别(Python代码,pyTorch框架,深度卷积网络模型,很容易替换为其它模型,带有GUI识别界面)_第1张图片

2. 每类照片展示

茶叶病害识别(Python代码,pyTorch框架,深度卷积网络模型,很容易替换为其它模型,带有GUI识别界面)_第2张图片

茶叶病害识别(Python代码,pyTorch框架,深度卷积网络模型,很容易替换为其它模型,带有GUI识别界面)_第3张图片

 茶叶病害识别(Python代码,pyTorch框架,深度卷积网络模型,很容易替换为其它模型,带有GUI识别界面)_第4张图片

 茶叶病害识别(Python代码,pyTorch框架,深度卷积网络模型,很容易替换为其它模型,带有GUI识别界面)_第5张图片

茶叶病害识别(Python代码,pyTorch框架,深度卷积网络模型,很容易替换为其它模型,带有GUI识别界面)_第6张图片

茶叶病害识别(Python代码,pyTorch框架,深度卷积网络模型,很容易替换为其它模型,带有GUI识别界面)_第7张图片

 茶叶病害识别(Python代码,pyTorch框架,深度卷积网络模型,很容易替换为其它模型,带有GUI识别界面)_第8张图片

 茶叶病害识别(Python代码,pyTorch框架,深度卷积网络模型,很容易替换为其它模型,带有GUI识别界面)_第9张图片

3整体文件夹

茶叶病害识别(Python代码,pyTorch框架,深度卷积网络模型,很容易替换为其它模型,带有GUI识别界面)_第10张图片

 

data文件夹存放的是未被划分训练集和测试集的原始照片

picture文件夹存放的是经hf.py对data文件夹处理后,生成的训练集和测试集照片

 

CNN.pth存放的是经train.py训练后的模型参数

GUI.py可以调用训练好的网络模型参数对多张照片连续进行识别

model.py 是存放模型的脚本,可以任意改为其它模型

predict.py是调用训练好的模型参数单个照片识别脚本,对单个 照片进行识别

4.经过60个epoch训练后,测试集平均准确率达到86%-90.08%的效果(用户如果计算机配置较高,可以增加epoch,效果也会更好),GUI界面识别效果和predict.py识别效果如视频所示茶叶病害识别(Python代码,pyTorch框架,深度卷积网络模型,很容易替换为其它模型,带有GUI识别界面)_哔哩哔哩_bilibili

 已经将代码和数据放在同一文件夹进行压缩,很容易进行运行代码,无需配置繁琐路径,对项目感兴趣的可以关注

import os
import json
 
import torch
from PIL import Image
from torchvision import transforms
import tkinter as tk
from tkinter import filedialog
 
from model import CNN
from PIL import ImageTk
 
#压缩包https://mbd.pub/o/bread/ZJ2VlZlp

你可能感兴趣的:(生物技术与深度学习结合,python,pytorch,开发语言)