一种农作物识别模型训练方法及相关设备

本申请提供了一种农作物识别模型训练方法，包括：获取有标注训练图像和无标注训练图像，将有标注训练图像输入第一农作物识别模型，获得第一预测结果，将无标注训练图像输入第二农作物识别模型，获得伪标签图像。根据第一预测结果和标注信息确定类别采样率。根据伪标签图像、无标注训练图像和类别采样率生成增强图像和增强图像的伪标签图像。将增强图像输入第一农作物识别模型，获得第二预测结果，并联合标注信息、第一预测结果、增强图像的伪标签图像，更新第一农作物识别模型的参数。该方法可以对于预测结果不佳的类别，生成增强图像来对模型进行训练，使得预测结果不佳的农作物类别能够有更多的机会参与模型的训练，提高模型的预测精确度。