[发明专利]一种基于深度迁移学习的茶叶病害识别方法

权利要求书

1.一种基于深度迁移学习的茶叶病害识别方法，其特征在于，该方法包括以下步骤：

训练阶段：

步骤1：获取多个品种茶叶及对应病害的图片，作为训练数据集；

步骤2：对训练数据集中的图片进行预处理，包括：随机翻转、随机剪裁、随机角度旋转、颜色抖动、添加噪声；

步骤3：构建图片分类模型；在DenseNet中引入SE Block，通过SE Block引入通道注意力机制，构建特征通道加权的SE-DenseNet网络模型；

步骤4：将训练数据集输入构建的SE-DenseNet网络模型中，通过迁移学习对SE-DenseNet网络模型进行训练，保存训练好的模型；

测试阶段：

步骤5：输入待识别的茶叶图片，通过训练得到的SE-DenseNet网络模型对茶叶图片中的病害进行分类识别，获得病害特征，通过分类器输出病害分类结果。

2.根据权利要求1所述的基于深度迁移学习的茶叶病害识别方法，其特征在于，所述步骤1中：

训练数据集包括公开数据集PlantVillage和自建数据集；数据集中包括不同茶叶品种的病害，病害包括茶白星病、茶轮斑病、茶煤病、茶圆赤星病和茶叶叶枯病。

3.根据权利要求1所述的基于深度迁移学习的茶叶病害识别方法，其特征在于，所述步骤2中：

1)随机翻转：依概率对图像进行水平、竖直翻转，病害相对位置不会发生改变，模拟相机不同角度的拍摄图像；

2)随机剪裁：对图像进行不同区域的随机截取，截取不同位置的图像以增加数据的多样性；

3)随机角度旋转：以一定的角度对图像进行旋转，获取不同角度的图像以丰富样本数据；

4)颜色抖动：修改图像的亮度、对比度、饱和度和颜色4个属性，以模拟不同光照条件下的拍摄图像；

5)添加噪声：对图像随机添加高斯噪声，得到不同清晰度的图像，以模拟不同的拍摄质量。

4.根据权利要求1所述的基于深度迁移学习的茶叶病害识别方法，其特征在于，所述步骤3中：

SENet由一系列SE block组成，每个SE block结构由Squeeze、Excitation、Reweigh三部分组成；在SE block中，Squeeze部分通过在特征图上执行全局平均池化得到当前特征图的全局压缩特征向量，Excitation通过两层全连接得到特征图中每个通道的权值，Reweigh将不同特征通道的权重乘以原来的特征图得到最终特征，最后将加权后的特征图作为下一层网络的输入；三部分映射关系分别为：

s＝F_ex(z，W)＝σ(g(z，W))＝σ(W₂δ(W₁z))

其中，C为通道数，W为特征图的宽度，H为特征图的高度，U_c代表每个特征通道，W₁和W₂为全连接层的权重矩阵，δ为全连接之间的激活函数ReLU，σ为归一化函数Sigmoid；

z_c为压缩操作后得到的压缩后的特征图；F_sq为压缩操作；i、j表示第c个二维矩阵U中的坐标；R^C表示特征通道的实数域；F_ex为激励操作；s表示通过激励操作得到的特征图权重；z为压缩后的特征图；F_scale为特征重标定操作；为特征重标定后的最终输出；s_c·u_c表示特征通道u_c与标量s_c的乘积。

SE-DenseNet模型结构中，如果神经网络有L层，每一层的输入都是前面所有层经过SENet重新矫正后的特征图输出的并集，其公式为：

其中，表示第l层经过SENet重标定后的特征图的输出，H_l表示非线性变换，表示0到l-1层经过SENet重标定后的输出特征图的归并操作。