代码运行要求: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是调用训练好的模型参数单个照片识别脚本,对单个 照片进行识别