农作物病虫害的模型训练方法

本发明公开了农作物病虫害的模型训练方法，模型训练方法包括：将农作物的原始图像输入人工智能模型的相互注意学习模块；基于相互注意学习模块输出原始图像的图像特征向量和注意力图，其中，注意力图降低原始图像的背景信息，并突显农作物信息；向人工智能模型的文本编码器输入病虫害的文本描述，得到文本特征向量；基于文本特征向量和图像特征向量获得原始图像中与病虫害相关的分类结果；根据分类结果与预设病虫害类别真值计算得出总损失函数，依据注意力图获得具有病虫害区域的新图像；利用新图像对人工智能模型进行增强训练，利用损失函数对人工智能模型的分类网络进行优化。本发明的技术方案能够更加有效的在农田中识别病虫害。