基于CNN的棉花不同病害叶识别(Python代码,pytorch框架,代码有详细中文注释,准确率在90%以上)
1.效果视频(训练过程:基于CNN模型的棉花不同病害叶识别(Python代码,pytorch框架)_哔哩哔哩_bilibili(为减小视频时长,epoch为30,准确率在85%左右,epoch为60后,稳定在90%以上),
GUI识别过程:棉花也病害识别GUI运行界面_哔哩哔哩_bilibili)

CNN模型介绍( CNN模型代码,可以替换为MobileNetV3Small, VGG16,AlexNet,ResNet18,GoogLeNet,很容易):

表 3-1 CNN完整网络参数
| 网络 | 名称 | 层类型 | 核尺寸/ 步长 | 核数量 | 激活 函数 |
| CNN 模块 | 卷积块1 | 卷积层 | 3*3/1 | 16 | ReLU |
| BN层 | — | — | — | ||
| 卷积块2 | 卷积层 | 3*3/1 | 32 | ReLU | |
| BN层 | — | — | — | ||
| 池化层1 | 最大池化层 | 2/2 | — | — | |
| 卷积块3 | 卷积层 | 3*3/1 | 64 | ReLU | |
| BN层 | — | — | — | ||
| 卷积块4 | 卷积层 | 3*3/1 | 128 | ReLU | |
| BN层 | — | — | — | ||
| 池化层2 | 自适应最大池化层 | — | — | — | |
| 分类器 | 全连接层1 | — | — | 256 | ReLU |
| 全连接层2 | — | — | 128 | ReLU | |
| 输出层 | 4 |
|
2.数据集介绍

如果想识别的时候呈现中文名称:
文件夹改为中文名字即可

-
Bacterial Blight(细菌性枯萎病):细菌性枯萎病是由细菌引起的棉花疾病,主要受害部位是棉花的叶子和茎。这种病害可以导致叶片枯萎、变色和腐烂,对棉花产量产生不利影响。
-
Curl Virus(卷叶病毒):卷叶病毒是一种病毒性病害,影响棉花植株。感染后,棉花叶片会卷曲并显示异常的颜色,这可能导致棉花生长不良和减产。
-
Fusarium Wilt(枯萎病):枯萎病是由一种真菌引起的棉花疾病。这种病害会导致棉花植株的叶子和茎部出现枯萎、变色和凋落的症状。枯萎病对棉花的生长和产量也造成了负面影响。
-
Healthy(健康):"Healthy" 表示没有任何上述病害或问题,棉花植株处于正常健康状态。
1.Bacterial Blight(细菌性枯萎病)文件夹(448张照片)

2. Curl Virus(卷叶病毒)文件夹(417张照片)

3. Fusarium Wilt(枯萎病)文件夹(419张照片)

4.健康文件夹(426张照片)
只对数据集感兴趣额的,可以关注棉花叶病害数据集_cotton insect pests 数据集-CSDN博客
整个文件夹的截图

背景照片是GUI程呈现的背景,可以替换
train.py是训练主程序,调用model.py里面写的CNN模型
model.py就是CNN模型
hf.py是对data文件夹里的原始数据进行分割训练集和测试集,生成的训练集和测试集保存在了piture文件夹(如果运行hf.py,需要重新删除piture文件夹)
CNN.pth就是train.py训练结束保存的模型参数。
class_indices.join可以被pycharm或者Spyder等Python语言编译器打开,里面是标签和对应的类别名称
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐


所有评论(0)