苹果叶病害识别(Python代码,pyTorch框架,预训练好的VGG16模型,也很容易替换为其它模型,带有GUI识别界面)

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

1.数据集介绍:

 Apple Scab类文件夹图片

 Black Rot类文件夹图片 

Cedar Apple Rust文件夹

 healthy文件夹

 2.整个项目

 

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

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

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

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

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

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

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

import os
import sys
import json

import torch
import torch.nn as nn
from torchvision import transforms, datasets
import torch.optim as optim
from tqdm import tqdm

# from model import GoogLeNet
from model import VGG16
#压缩包https://mbd.pub/o/bread/ZJ2Vmpxt

猜你喜欢

转载自blog.csdn.net/qq_40840797/article/details/132454332
今日推荐