摘要
本发明公开了一种改进型植物病害数据增强方法,包括步骤S1,引入植物病害数据集和潜在扩散模型,将各植物图片作为潜在扩散模型的输入,将预先设置的多组提示文本作为潜在扩散模型的文本提示条件;步骤S2,利用ControlNet算法提取植物图片中的图像特征,并将图像特征作为指导调整潜在扩散模型的输出,以对潜在扩散模型初步优化;步骤S3,利用LORA优化算法维持潜在扩散模型的原始权重矩阵不变,更新潜在扩散模型的低秩分解,以对潜在扩散模型再次优化;步骤S4,利用文本嵌入指导模型提取各提示文本中的文本特征,将文本特征嵌入潜在扩散模型的去噪网络中,以对潜在扩散模型进一步优化。本发明提升生成的植物病害数据的质量。