模型训练方法技术

本申请公开了一种模型训练方法，涉及深度学习技术领域。方法包括：获取待训练人脸图像降噪模型以及训练样本集，训练样本集包括多组图像对，每组图像对包括样本人脸图像和参考人脸图像；待训练人脸图像降噪模型用于对样本人脸图像进行降噪处理，获得降噪后图像；对样本人脸图像进行语义分割，获得至少一个语义分割区域；基于参考人脸图像和降噪后图像之间各语义分割区域的结构损失，构建待训练人脸图像降噪模型的第一损失函数；利用训练样本集和第一损失函数训练待训练人脸图像降噪模型，获得目标图像降噪模型。本申请可使得训练得到的人脸图像降噪网络可以更好的在降噪处理过程中保留细节特征。留细节特征。留细节特征。

【技术实现步骤摘要】
模型训练方法


[0001]本申请涉及深度学习
，尤其涉及一种模型训练方法。

技术介绍

[0002]在进行人脸图像信号处理时，可以使用基于深度学习的图像去噪方法进行图像降噪，即将待处理人脸图像输入至图像降噪网络模型中进行降噪处理。
[0003]但是，相关技术中，图像降噪模型容易将人脸图像中不同目标物之间的边界细节特征当成噪点去除，如图像降噪模型对低照度场景下的人脸图像进行降噪处理时，容易将人脸五官边界细节当成噪点去除，从而影响最终的图像信号处理质量。

技术实现思路

[0004]本申请的主要目的在于提供一种模型训练方法，旨在解决图像降噪模型降噪不准确的技术问题。
[0005]为实现上述目的，本申请提供一种模型训练方法，方法包括：获取待训练人脸图像降噪模型以及训练样本集，训练样本集包括多组图像对，每组图像对包括样本人脸图像和参考人脸图像；待训练人脸图像降噪模型用于对样本人脸图像进行降噪处理，获得降噪后图像；对样本人脸图像进行语义分割，获得至少一个语义分割区域；基于参考人脸图像和降噪后图像之间各语义分割区域的结构损失，构建待训练人脸图像降噪模型的第一损失函数；利用训练样本集和第一损失函数训练待训练人脸图像降噪模型，获得目标图像降噪模型。
[0006]在本申请可能的一实施例中，基于参考人脸图像和降噪后图像之间各语义分割区域的结构损失，构建待训练人脸图像降噪模型的第一损失函数，包括：基于参考人脸图像和降噪后图像之间各语义分割区域的结构损失，以及各语义分割区域的区域权重，构建待训练人脸图像降噪模型的第一损失函数。
[0007]在本申请可能的一实施例中，基于参考人脸图像和降噪后图像之间各语义分割区域的结构损失，构建待训练人脸图像降噪模型的第一损失函数，包括：确定待训练人脸图像降噪模型的第一参考损失函数分量；基于参考人脸图像和降噪后图像之间各语义分割区域的结构损失，构建结构损失函数分量；基于第一参考损失函数分量，第一参考损失函数分量的第一权重、结构损失函数分量以及结构损失函数分量的第二权重，构建第一损失函数。
[0008]在本申请可能的一实施例中，获取待训练人脸图像降噪模型以及训练样本集，包括：获取待训练图像降噪插值网络模型以及训练样本集；图像降噪插值网络模型包括
依次连接的待训练人脸图像降噪模型和图像插值模型，样本人脸图像为RAW原始人脸图像，参考人脸图像为彩色人脸图像，且图像插值模型用于对降噪后图像进行光谱上采样插值重建，获得重建彩色人脸图像；利用训练样本集和第一损失函数训练待训练人脸图像降噪模型，获得目标图像降噪模型，包括：基于重建彩色人脸图像和参考人脸图像，构建图像插值模型的第二损失函数；利用训练样本集、第一损失函数和第二损失函数，训练待训练图像降噪插值网络模型，获得目标图像降噪插值网络模型。
[0009]在本申请可能的一实施例中，基于重建彩色人脸图像和参考人脸图像，构建图像插值模型的第二损失函数，包括：确定图像插值模型的第二参考损失函数分量；基于重建彩色人脸图像和参考人脸图像之间的高低频损失函数分量、高低频损失函数分量对应的第三权重、第二参考损失函数分量以及第二参考损失函数分量的第四权重，构建第二损失函数。
[0010]在本申请可能的一实施例中，利用训练样本集、第一损失函数和第二损失函数，训练待训练图像降噪插值网络模型，获得目标图像降噪插值网络模型，包括：基于第一损失函数、第一损失函数的第五权重、第二损失函数以及第二损失函数的第六权重，构建联合损失函数；利用训练样本集和联合损失函数，训练待训练图像降噪插值网络模型，获得目标图像降噪插值网络模型。
[0011]在本申请可能的一实施例中，待训练图像降噪插值网络模型包括通道分离模块、输出层以及通道分离模块和输出层之间并排的多个通道；每个通道包括依次连接的待训练人脸图像降噪模型和图像插值模型，且图像插值模型用于对输入的降噪后图像进行插值和上采样，输出上采样图像，上采样图像的尺寸和样本人脸图像的尺寸相同；通道分离模块用于接收样本人脸图像，基于多个预设尺寸对样本人脸图像分别进行下采样，获得多个通道输入图像，将通道输入图像输入至对应的通道；其中，多个预设尺寸彼此不同，且多个预设尺寸与多个通道一一对应；输出层用于接收各个通道输出的上采样图像，将多个上采样图像融合得到重建彩色人脸图像。
[0012]在本申请可能的一实施例中，利用训练样本集和联合损失函数，训练待训练图像降噪插值网络模型，获得目标图像降噪插值网络模型，包括：基于训练步数和各个通道的序号，构建各个通道的训练步数相关度；其中，通道的序号与预设尺寸成负相关，当训练步数小于预设收敛步数时，训练步数相关度和序号呈正相关，当训练步数大于预设收敛步数时，训练步数相关度和序号呈负相关；将训练步数相关度作为对应的通道的通道权重；基于各个通道的联合损失函数和各个通道的通道权重，构建网络总体损失函数；利用训练样本集和网络总体损失函数，训练待训练图像降噪插值网络模型，获得目标图像降噪插值网络模型。
[0013]在本申请可能的一实施例中，基于训练步数和各个通道的序号，构建各个通道的训练步数相关度，包括：基于训练步数、各个通道的序号和公式一，构建各个通道的训练步数相关度；公式一为：；其中，，满足:,为第个通道的通道权重，为训练步数，为预设收敛步数，为大于或者等于1的常数。
[0014]在本申请可能的一实施例中，利用训练样本集和网络总体损失函数，训练待训练图像降噪插值网络模型，获得目标图像降噪插值网络模型，包括：将网络总体损失函数中的第五权重赋值为1，第六权重赋值为0，获得第一状态网络总体损失函数；利用训练样本集训练待训练图像降噪插值网络模型，并使用第一状态网络总体损失函数更新待训练图像降噪插值网络模型的模型参数，直至第一状态网络总体损失函数收敛，获得第一阶段训练模型；将第五权重赋值为0，第六权重赋值为1，获得第二状态网络总体损失函数；利用训练样本集训练第一阶段训练模型，并使用第二状态网络总体损失函数更新第一阶段训练模型的模型参数，直至第二状态网络总体损失函数收敛，获得第二阶段训练模型；确定网络总体损失函数中的第五权重的第一初始赋值和第六权重赋值的第二初始赋值，获得第三状态网络总体损失函数；其中，第一初始赋值和第二初始赋值均大于1，且两者的和等于1；利用训练样本集训练第二阶段训练模型，并使用第三状态网络总体损失函数更新第五权重和第六权重，直至第二阶段训练模型收敛得到目标图像降噪插值网络模型。
[0015]在本申请可能的一实施例中，确定网络总体损失函数中的第五权重的第一初始赋值和第六权重赋值的第二初始赋值，获得第三状态网络总体损失函数之前，方法还包括：确定当前的第二阶段训练模型；根据损失函数交替训练规则，从第一状态网络总体损失函数和第二状态网络总体损失函数中确定出当前训练损失函数；利用训练样本集训练当前的第二阶段训练模型，并使用当前训练损失函数更新当前的第二阶段训练模型的模型参数，直至当前训练损失函数收敛；返回执行确定当前的第二阶段训练模型，直至第一状态网络总体损失函数本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种模型训练方法，其特征在于，所述方法包括：获取待训练人脸图像降噪模型以及训练样本集，所述训练样本集包括多组图像对，每组所述图像对包括样本人脸图像和参考人脸图像；所述待训练人脸图像降噪模型用于对所述样本人脸图像进行降噪处理，获得降噪后图像；对所述样本人脸图像进行语义分割，获得至少一个语义分割区域；基于所述参考人脸图像和所述降噪后图像之间各所述语义分割区域的结构损失，构建所述待训练人脸图像降噪模型的第一损失函数；利用所述训练样本集和所述第一损失函数训练所述待训练人脸图像降噪模型，获得目标图像降噪模型。2.根据权利要求1所述的模型训练方法，其特征在于，所述基于所述参考人脸图像和所述降噪后图像之间各所述语义分割区域的结构损失，构建所述待训练人脸图像降噪模型的第一损失函数，包括：基于所述参考人脸图像和所述降噪后图像之间各所述语义分割区域的结构损失，以及各所述语义分割区域的区域权重，构建所述待训练人脸图像降噪模型的第一损失函数。3.根据权利要求1所述的模型训练方法，其特征在于，所述基于所述参考人脸图像和所述降噪后图像之间各所述语义分割区域的结构损失，构建所述待训练人脸图像降噪模型的第一损失函数，包括：确定所述待训练人脸图像降噪模型的第一参考损失函数分量；基于所述参考人脸图像和所述降噪后图像之间各所述语义分割区域的结构损失，构建结构损失函数分量；基于所述第一参考损失函数分量，所述第一参考损失函数分量的第一权重、所述结构损失函数分量以及所述结构损失函数分量的第二权重，构建所述第一损失函数。4.根据权利要求1至3任一项所述的模型训练方法，其特征在于，所述获取待训练人脸图像降噪模型以及训练样本集，包括：获取待训练图像降噪插值网络模型以及训练样本集；所述图像降噪插值网络模型包括依次连接的所述待训练人脸图像降噪模型和图像插值模型，所述样本人脸图像为RAW原始人脸图像，所述参考人脸图像为彩色人脸图像，且图像插值模型用于对所述降噪后图像进行光谱上采样插值重建，获得重建彩色人脸图像；所述利用所述训练样本集和所述第一损失函数训练所述待训练人脸图像降噪模型，获得目标图像降噪模型，包括：基于所述重建彩色人脸图像和所述参考人脸图像，构建所述图像插值模型的第二损失函数；利用所述训练样本集、所述第一损失函数和所述第二损失函数，训练所述待训练图像降噪插值网络模型，获得目标图像降噪插值网络模型。5.根据权利要求4所述的模型训练方法，其特征在于，所述基于所述重建彩色人脸图像和所述参考人脸图像，构建所述图像插值模型的第二损失函数，包括：确定所述图像插值模型的第二参考损失函数分量；基于所述重建彩色人脸图像和所述参考人脸图像之间的高低频损失函数分量、所述高低频损失函数分量对应的第三权重、所述第二参考损失函数分量以及所述第二参考损失函
数分量的第四权重，构建所述第二损失函数。6.根据权利要求4所述的模型训练方法，其特征在于，所述利用所述训练样本集、所述第一损失函数和所述第二损失函数，训练所述待训练图像降噪插值网络模型，获得目标图像降噪插值网络模型，包括：基于所述第一损失函数、所述第一损失函数的第五权重、所述第二损失函数以及所述第二损失函数的第六权重，构建联合损失函数；利用所述训练样本集和所述联合损失函数，训练所述待训练图像降噪插值网络模型，获得目标图像降...

【专利技术属性】
技术研发人员：沈晰强，傅斌，
申请(专利权)人：杭州海康威视数字技术股份有限公司，
类型：发明
国别省市：