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

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐