摘要
本发明公开了一种模型训练、谷物图像识别方法、存储介质和电子设备,涉及人工智能技术领域,所述模型训练方法包括:获得带有多个任务标签的谷物图像数据;将所述谷物图像数据输入预设的神经网络模型,得到所述神经网络模型输出的处理结果;所述处理结果包括:所述谷物图像数据的第一图像特征,以及所述谷物图像数据的第二图像特征与各任务标签对应的各第一文本特征之间的相似度;基于所述第一图像特征确定对所述谷物图像数据进行数据处理的第一损失,并基于所述相似度确定对所述谷物图像数据进行数据处理的第二损失;基于所述第一损失和所述第二损失调整所述神经网络模型的模型参数。应用本发明实施例提供的方案能够提高模型训练效率。
技术关键词
模型训练方法
神经网络模型
图像编码器
图像识别方法
标签
数据
文本编码器
分支
样本
共享图像
编码器模块
电子设备
人工智能技术
处理器
优化器
参数
可读存储介质