【毕业设计选题】基于深度学习的中草药检测识别系统 人工智能 python 计算机视觉 YOLO

目录

一、课题背景与意义

        叶片是中草药植物的重要器官，也是区分不同植物的重要参照，大多数中草药的叶片都趋于扁平 化，便于被摄录设备采集并进行数字化存储。传统的中草药叶片识别往往需要人工采集大量的原始样 本，再进行人工分析和研究，需要研究者具有相当丰富的中医药知识储备和经验，整个过程耗时耗力且 识别率难以得到保证。在研究中草药植物叶片分类过程中，借助人工智能算 法的优势，开展基于深度神经网络的图像分类实验，进一步提高识别准确率。

二、算法理论技术

2.1 卷积神经网络

        CNN（卷积神经网络）是一种多层网络模型，用于识别二维形状。它具有对平移、比例缩放、倾斜和其他形式的变形具有高度不变性的特点。CNN采用局部连接和局部感受野的结构，与传统的全连接神经网络不同。从输入层开始，每一层接收上一层的输入，一直到输出层，形成一个有向环图。CNN通过卷积运算和局部池化操作来提取特征，使得得到的特征图逐渐变得复杂。每个图由多个神经元构成，多个图构成一个层。这些图充当过滤器的作用，通过计算输入图像的卷积值并与特征匹配，得到适当的特征，并将其输入到下一层结构中。

        CNN框架通过卷积和池化操作对输入的中草药叶片图像进行特征提取和分类。图像经过多层的卷积和池化操作后，最终得到分类结果。该方法利用CNN的优势，提高了中草药叶片识别的准确率和效率。

代码如下（示例）：

import torch
import torch.nn as nn

class CNN(nn.Module):
    def __init__(self, num_classes):
        super(CNN, self).__init__()
        self.conv1 = nn.Conv2d(3, 16, kernel_size=3, stride=1, padding=1)
        self.relu = nn.ReLU()
        self.maxpool = nn.MaxPool2d(kernel_size=2, stride=2)
        self.conv2 = nn.Conv2d(16, 32, kernel_size=3, stride=1, padding=1)
        self.fc = nn.Linear(32 * 16 * 16, num_classes)

    def forward(self, x):
        x = self.conv1(x)
        x = self.relu(x)
        x = self.maxpool(x)
        x = self.conv2(x)
        x = self.relu(x)
        x = self.maxpool(x)
        x = x.view(x.size(0), -1)
        x = self.fc(x)
        return x

2.2 AlexNet模型

        AlexNet是一种深度卷积神经网络模型，通过深层的卷积和池化层堆叠、非线性激活函数、局部响应归一化和Dropout正则化等创新设计，实现了对大规模图像数据集的高效分类。它在计算机视觉领域取得了重要突破，开创了深度学习在图像识别任务上的先河，并对后续网络模型的发展和应用产生了深远影响。

        对原始的AlexNet模型进行了改进和优化。在第三层和第四层特征提取能力最强的基础上，为了进一步提高分类的准确度，插入了一层新的第五层。该新层与原第四层结构一致，但通过增加卷积操作来提取更多有效特征。经过多次实验和探索，改进后的模型命名为AlexNet_En，并且在分类精确度上优于原始的AlexNet模型。AlexNet_En模型的结构，其中第五层以虚线框表示。这种改进和优化有助于进一步提升模型性能和特征提取的准确度。

代码如下（示例）：

import torch
import torch.nn as nn

class AlexNet_En(nn.Module):
    def __init__(self, num_classes):
        super(AlexNet_En, self).__init__()
        self.features = nn.Sequential(
            nn.Conv2d(3, 64, kernel_size=11, stride=4, padding=2),
            nn.ReLU(inplace=True),
            nn.MaxPool2d(kernel_size=3, stride=2),
            nn.Conv2d(64, 192, kernel_size=5, padding=2),
            nn.ReLU(inplace=True),
            nn.MaxPool2d(kernel_size=3, stride=2),
            nn.Conv2d(192, 384, kernel_size=3, padding=1),
            nn.ReLU(inplace=True),
            nn.Conv2d(384, 384, kernel_size=3, padding=1),
            nn.ReLU(inplace=True),
            nn.Conv2d(384, 256, kernel_size=3, padding=1),
            nn.ReLU(inplace=True),
            nn.MaxPool2d(kernel_size=3, stride=2),
        )
        self.classifier = nn.Sequential(
            nn.Dropout(),
            nn.Linear(256 * 6 * 6, 4096),
            nn.ReLU(inplace=True),
            nn.Dropout(),
            nn.Linear(4096, 4096),
            nn.ReLU(inplace=True),
            nn.Linear(4096, num_classes),
        )

    def forward(self, x):
        x = self.features(x)
        x = x.view(x.size(0), 256 * 6 * 6)
        x = self.classifier(x)
        return x

三、模型训练

3.1 数据处理

        通过python爬虫算法爬取百度图片，中文关键字为“苍耳”（xanthium si biricum）、“金银花”（honeysuckle）、“薄荷”（mint）、“藿香”（patchouli）、“白花蛇舌草”（hedyotis diffusa）共5类中药植物图片，每类挑选其中600张，共3000张图片组成数据集，其中2400张作为训练集，300张作为验证集，300张作为测试集，各占总图像数量的80%、10%、10%，将该数据集命名为HERB3K。

        为了解决数据集HERB3K样本数量较少可能影响训练的准确率问题，拟采用以下3种方法对数据集HERB3K进行数据增扩。关于图像增噪有许多算法可以实现，本研究采用加性零均值高斯噪声方法为图像增加相应的噪点。增噪的方式是为图像上每个点的灰度值加上一个噪声值，噪声值的产生方式是使用Box-Muller算法生成高斯噪声。

代码如下（示例）：

import numpy as np
import cv2

def add_gaussian_noise(image, mean=0, std=1):
    # 生成相应大小的随机高斯噪声
    noise = np.random.normal(mean, std, image.shape).astype(np.uint8)
    
    # 将噪声添加到原始图像上
    noisy_image = cv2.add(image, noise)
    
    return noisy_image

3.2 实验环境

        实验基于Caffe平台搭建，Caffe是一款基于C++/CUDA，支持命令行、Matlab和Python接口的深度学习框架，该框架利用MKL、OpenBLAS、cuBLAS等矩阵运算库加速计算、同时支持GPU加速，具有例程脚本丰富、入门快等特点。实验软硬件环境为Mac OS操作系统、Intel Core i5 1.6 G CPU、8G内存。

3.3 结果分析

        默认的AlexNet模型使用的是1000类的ImageNet数据集，而本研究只采集了5种中草药图像，因此需要修改每组实验模型参数num_output=5。此外，在训练过程中多次测试发现每组进行了300次左右迭代后准确率及损失值基本趋于平稳。

        改进的AlexNet模型在数据增扩下具有更高的训练准确率。然而，相比于原始的AlexNet模型在数据增扩下所获得的准确率提高，改进的模型的提升幅度较小。基于改进的AlexNet模型的训练结果普遍优于原始模型。结合数据增扩和改进的AlexNet模型能够最大程度地提高图像分类的准确率。

        通过对比和分析改进的AlexNet_En网络模型与原始AlexNet网络模型以及数据增扩后的数据集与原始数据集，展示出了较好的分类效果。然而，从损失值的角度来看，改进后的AlexNet模型的损失值普遍高于原始AlexNet模型。此外，改进后的AlexNet_En模型的训练时间也较原始AlexNet模型长，因为改进后的模型层数较多，复杂度较高，对硬件要求也更高。同时，随着数据集的增大，AlexNet_En对分类性能的提升幅度变小，而在HERB3K这样的小数据集上有较大的准确率提升，这表明存在过拟合问题。这些问题将成为未来模型改进和提升的方向。

最后