[发明专利]一种模型训练以及图像处理方法及装置

说明书

本说明书公开了一种模型训练以及图像处理方法及装置，可获取不同分辨率的各第一图像及对应的各第二图像，并根据各图像的对应关系以及各图像中包含的内容，确定各训练样本及标注。之后，将各训练样本分别输入第一特征提取层以及超分辨率网络层，确定低分辨率特征、第三图像以及分辨率转化特征，并基于该第三图像确定高分辨率特征，将低分辨率特征、分辨率转化特征以及高分辨率特征输入识别网络层，确定识别结果。最后，根据第三图像、识别结果以及该训练样本的标注确定最终损失，并以最小化最终损失为目标，调整该待训练的识别模型中的模型参数。通过低分辨率特征、分辨率转化特征以及高分辨率特征，共同进行图像识别，使图像识别结果更加准确。

技术领域

本申请涉及机器学习技术领域，尤其涉及一种模型训练以及图像处理方法及装置。

背景技术

图像识别是一种常见的图像处理方法，广泛应用于各领域中，如，人脸识别以及智能机器人的障碍物识别等。

一般的，图像识别的成功率以及准确率与待识别图像的分辨率相关，待识别图像的分辨率越高，图像识别的成功率越高，图像识别准确率也越高。

但是，基于图像采集环境以及图像采集设备等因素的影响，当图像采集环境的光线较暗、图像采集设备的像素较差时，所采集的待识别图像的清晰度也较差，也就是说，图像分辨率较低，进一步导致图像识别的准确率较低。

发明内容

本说明书实施例提供一种模型训练以及图像处理方法及装置，用于部分解决现有技术中存在的问题。

本说明书实施例采用下述技术方案：

本说明书提供的一种模型训练方法，待训练的识别模型至少包含第一特征提取层、超分辨率网络层、第二特征提取层以及识别网络层；

所述方法包括：

获取若干张高分辨率的第一图像及其对应的低分辨率的第二图像，所述第一图像中包含目标物体；

根据确定出的各第二图像，确定各训练样本，并根据各第二图像对应的第一图像，确定各训练样本的第一标注，以及根据各第二图像中包含的目标物体的属性，确定各训练样本的第二标注；

针对每个训练样本，将该训练样本分别输入所述第一特征提取层，确定输出的低分辨率特征，以及输入所述超分辨率网络层，确定输出的高分辨率的第三图像以及分辨率转化特征；

将所述第三图像输入所述第二特征提取层，确定输出的高分辨率特征；

将确定出的低分辨率特征、分辨率转化特征以及高分辨率特征输入所述识别网络层，确定输出的识别结果；

根据通过所述第三图像以及该训练样本的第一标注确定出的第一损失，以及通过所述识别结果以及该训练样本的第二标注确定出的第二损失，确定最终损失，并以最小化所述最终损失为目标，调整所述待训练的识别模型中的模型参数，所述识别模型用于根据低分辨率图像确定识别结果。

可选地，所述识别模型还包含第三特征提取层；

将确定出的低分辨率特征、分辨率转化特征以及高分辨率特征输入所述识别网络层，具体包括：

将所述分辨率转化特征输入所述第三特征提取层，确定输出的转化深层特征；

将确定出的低分辨率特征、高分辨率特征以及转化深层特征输入所述识别网络层。

可选地，所述识别模型还包含特征融合层；

将确定出的低分辨率特征、分辨率转化特征以及高分辨率特征输入所述识别网络层，具体包括：

将确定出的低分辨率特征、分辨率转化特征以及高分辨率特征输入所述特征融合层，确定输出的融合特征；

将确定出的融合特征输入所述识别网络层。