摘要
本申请提出一种模型训练方法、图像识别方法、设备及存储介质,包括:将第一样本图像输入第一图像识别模型,输出第一预测分类结果和第一预测掩膜;将目标特征图像输入至退化预测网络,输出退化参数预测结果;基于第一预测分类结果和第一预测掩膜,计算第一损失函数值;以及基于退化参数和退化参数预测结果计算第二损失函数值;基于上述两个损失函数调整第一图像识别模型的模型参数,训练第一图像识别模型。本申请实施例通过预测分类结果和预测掩膜的损失函数以及退化参数损失函数优化网络参数,以便在模型训练的过程中同时注意分类和掩膜的图像特征,以及能够提取不同退化程度下的可还原至退化前的图像特征,从而提高模型的识别精准度。
技术关键词
图像识别模型
模型训练方法
图像识别方法
网络
掩膜
样本
模糊参数
引入注意力机制
损失函数优化
噪声信息
处理器
对象
可读存储介质
上采样
存储器
训练集
电子设备
计算机
系统为您推荐了相关专利信息
涂层性能检测方法
磁阻传感器
信号特征值
涡流探头
反演模型
电网故障定位方法
电气运行设备
卷积网络模型
节点特征
神经网络模型
数据处理功能
数据处理方法
计算机程序产品
计算机设备
芯片
问答模型
样本
计算机程序产品
网络
可读存储介质