苹果叶病害识别(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是调用训练好的模型参数单个照片识别脚本,对单个 照片进行识别
更多推荐
所有评论(0)