一种基于神经网络模型的柑橘病虫害检测方法技术

技术编号:33544906 阅读:28 留言:0更新日期:2022-05-21 10:01
本发明专利技术公开了一种基于神经网络模型的柑橘病虫害检测方法,包括以下步骤:步骤1、获取柑橘病虫害图像作为数据集;步骤2、将数据集分为训练集和测试集,并进行预处理;步骤3、构建Contextual Swin Transformer模型,对模型进行训练得到最优配置参数;步骤4、通过参数为最优配置参数的模型对待识别病虫害种类的柑橘病图像进行识别,得到病虫害识别结果。本发明专利技术可降低计算量,提高检测效率和准确性。提高检测效率和准确性。提高检测效率和准确性。

【技术实现步骤摘要】
Swin Transformer 模型,由Contextual Swin Transformer 模型输出最终的病虫害种类识别结果。
[0006]进一步的步骤2中,按7:3的比例将数据集划分为训练集和测试集。
[0007]进一步的,步骤2中所述的预处理包括数据广增、图像填充以及具有层级的滑窗操作处理。
[0008]进一步的,所述数据广增预处理依次包括随机翻转、缩放、随机裁剪、归一化。
[0009]进一步的,所述训练集中数据进行数据广增时,以0.5的概率进行随机翻转,并从若干种尺度中随机挑选一种以对训练集中数据进行缩放。
[0010]进一步的步骤3中,由Contextual Swin Transformer 模型基于自注意力机制将输入的图像根据大小windows size划分分成n个不重叠windows,得到=( ,...,),x表示根据windows size划分后得到的向量;Contextual Swin Transformer 模型中第i个自注意力机制的输出,由输入与对应的query参数矩阵、key参数矩阵相乘再经过SoftMax函数,最后与本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于神经网络模型的柑橘病虫害检测方法,其特征在于,基于Contextual Swin Transformer 模型实现检测,包括以下步骤:步骤1、获取多种柑橘病虫害的多张图像作为数据集;步骤2、将数据集分为训练集和测试集,并分别对训练集和测试集中的数据进行预处理;步骤3、构建Contextual Swin Transformer 模型,并将所述训练集输入至Contextual Swin Transformer 模型,以对Contextual Swin Transformer 模型进行训练;每次训练时利用Contextual Swin Transformer 模型中的自注意力机制,将Contextual Swin Transformer 模型的输入映射成query向量、key向量、value向量,由此使三个向量中的信息传递至Contextual Swin Transformer 模型的相对位置编码中;每次训练后由Contextual Swin Transformer 模型输出识别的病虫害种类作为输出结果,并将输出结果与测试集进行误差计算,然后基于误差计算结果调整Contextual Swin Transformer 模型的配置参数,由此经过多次训练,得到误差计算结果符合预期时的Contextual Swin Transformer 模型的最优配置参数;步骤4、设置Contextual Swin Transformer 模型的配置参数为所述最优配置参数,将待识别病虫害种类的柑橘病图像输入至参数调整为最优配置参数的Contextual Swin Transformer 模型,由Contextual Swin Transformer 模型输出最终的病虫害种类识别结果。2.根据权利要求1所述的一种基于神经网络模型的柑橘病虫害检...

【专利技术属性】
技术研发人员:吴琪吴云志
申请(专利权)人:安徽农业大学
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1