[发明专利]一种基于Transformer的农作物叶片病害检测方法

权利要求书

1.一种基于Transformer的农作物叶片病害检测方法，其特征在于，包括以下步骤：

S1、获取n幅用于模型训练的农作物叶片图像，标记农作物叶片图像中的病害区域，得到完成标记的农作物叶片图像的样本集；

S2、构建初始的叶片病害检测模型，所述初始的叶片病害检测模型包括编码器和解码器，编码器采用Transformer结构进行高层特征提取，解码器采用通道注意力机制对编码器Transformer输出特征图进行特征融合；

S3、利用步骤S1的样本集训练初始的叶片病害检测模型，从而得到训练完成的叶片病害检测模型；

S4、利用训练完成的叶片病害检测模型对待检测的农作物叶片图像进行检测，得到农作物叶片图像中农作物叶片病害检测结果。

2.根据权利要求1所述的一种基于Transformer的农作物叶片病害检测方法，其特征在于，步骤S1中获取所述标记农作物叶片图像的方法具体为：

S11、获取n幅用于模型训练的农作物叶片图像，其中农作物叶片图像具有RGB三个通道；所述农作物叶片图像包括健康叶片的农作物叶片图像和病害叶片的农作物叶片图像；

S12、标记每一幅病害叶片农作物叶片图像中的病害区域和病害区域的病害类型，标记每一幅健康叶片农作物叶片图像中的健康叶片，根据所述标记数据导出掩码图像，每一幅所述农作物叶片图像对应一幅掩码图像；其中掩码图像和农作物叶片图像长宽相同，掩码图像为单通道的灰度图，掩码图像中每个像素点均对应有病害类型或健康叶片的标签。

3.根据权利要求2所述的一种基于Transformer的农作物叶片病害检测方法，其特征在于，步骤S2所述叶片病害检测模型的编码器包括：输入层、卷积层和Transformer层，具体为：

输入层输入农作物叶片图像，并将农作物叶片图像传输到第一卷积层；农作物叶片图像的尺寸为H×W×3，其中，H和W分别为农作物叶片图像的长宽，3表示农作物叶片图像具有RGB三个通道；

农作物叶片图像经过第一卷积层，得到第一编码特征图，第一编码特征图的尺寸为H×W×24；

第一编码特征图首先输入池化层进行池化处理，然后经过第二卷积层处理后得到第二编码特征图，第二编码特征图的尺寸为

农作物叶片图像经过池化层进行四倍下采样处理，然后输入第一Transformer层得到第三编码特征图，第三编码特征图尺寸为

第三编码特征图输入第二Transformer层得到第四编码特征图，第四编码特征图尺寸为

第四编码特征图输入第三Transformer层得到第五编码特征图，第五编码特征图尺寸为

第五编码特征图输入第四Transformer层得到第六编码特征图，第六编码特征图尺寸为

4.根据权利要求3所述的一种基于Transformer的农作物叶片病害检测方法，其特征在于，所述第一Transformer层、第二Transformer层、第三Transformer层和第四Transformer层的具体结构为：

所述第一Transformer层包括：池化层、区块划分操作模块、线性编码操作模块和Transformer Block；农作物叶片图像依次经过池化层、区块划分操作模块、线性编码操作模块和Transformer Block得到第三编码特征图；

第二Transformer层、第三Transformer层和第四Transformer层均为下采样操作模块和Transformer Block依次连接；

其中，所述Transformer层中的下采样操作模块均是将编码特征图划分为四个区域，然后在深度方向进行拼接变成[0,1,2,3]，再经过线性变换变成[4,5]。