图像处理模型训练方法、装置、设备及存储介质制造方法及图纸

本申请公开了一种图像处理模型训练方法、装置、设备及存储介质，属于人工智能技术领域。本申请实施例中，一方面，引入了遮挡模式，通过确定该样本人脸图像的遮挡模式，并与该样本人脸图像所标注的目标遮挡模式做对比，训练图像处理模型确定出更准确的遮挡指示信息，提升人脸识别的准确性，提升了图像处理模型的鲁棒性。另一方面，该图像处理模型无需借助外部网络单独进行遮挡区域的预测过程以及遮挡区域对应特征的字典查询过程，能够端到端地对图像进行处理，显著地减少了计算量，提升设备的运行速度，也减少了图像处理模型的大小，且不受外部网络因素影响，准确性得到了显著提升。

【技术实现步骤摘要】
图像处理模型训练方法、装置、设备及存储介质
本申请涉及人工智能
，特别涉及一种图像处理模型训练方法、装置、设备及存储介质。
技术介绍
随着计算机技术的发展，人工智能应用在各个领域，基于人工智能来代替人的工作，能够大大提高业务处理效率。在图像处理方面，能够对图像处理模型进行训练，得到训练好的图像处理模型，将待分类的图像输入训练好的图像处理模型即可得到相应的处理结果。目前，对带遮挡的人脸图像的图像处理模型训练方法通常是先训练一种网络，建立起将人脸被遮挡区域映射为受影响大的图像特征的字典，然后获取样本人脸图像后，由图像处理模型进行特征提取，然后借助外部训练的网络检测样本人脸图像中人脸被遮挡区域，通过查询上面的字典，确定出受影响大的图像特征后，再由图像处理模型将这部分图像特征去除后进行人脸识别，再根据识别结果进行训练。上述方法需要图像处理模型需要借助外部网络检测人脸遮挡区域，确定受影响大的图像特征，设备的计算量非常大，运行速度较慢，且建立字典的过程较为繁琐，耗时和训练成本较高。另外，外部网络检测的精度也直接影响到该图像处理模型处理结果，如果外部网络检测不准，该图像处理模型处理图像的准确性则会受到很大影响。
技术实现思路
本申请实施例提供了一种图像处理模型训练方法、装置、设备及存储介质，能够提高图像处理模型处理图像的准确性和鲁棒性，显著地减少计算量，提升了设备的运行速度，且训练得到的图像处理模型的大小较小。所述技术方案如下：一方面，提供了一种图像处理模型训练方法，所述方法包括：基于图像处理模型，获取样本人脸图像的预测识别结果和遮挡指示信息，所述遮挡指示信息用于指示所述样本人脸图像的受人脸被遮挡区域影响的图像特征；基于所述预测识别结果和所述样本人脸图像所标注的目标识别结果，获取识别误差；基于所述样本人脸图像的遮挡指示信息，对所述样本人脸图像的遮挡模式进行分类，得到预测遮挡模式，所述遮挡模式用于指示所述样本人脸图像中人脸被遮挡区域的位置以及尺寸；基于所述预测遮挡模式和所述样本人脸图像所标注的目标遮挡模式，获取分类误差；根据所述识别误差和所述分类误差，对所述图像处理模型的模型参数进行更新。在一种可能实现方式中，所述基于所述遮挡指示信息，去除所述图像特征中所述人脸被遮挡区域的图像特征，得到目标图像特征，包括：将所述图像特征与所述遮挡指示信息相乘，得到所述目标图像特征。一方面，提供了一种图像处理方法，所述方法包括：对待识别的人脸图像进行特征提取，得到所述人脸图像的图像特征；基于所述图像特征，确定对应的遮挡指示信息，所述遮挡指示信息用于指示所述人脸图像中各区域的人脸是否被遮挡；根据所述图像特征和所述遮挡指示信息，获取目标图像特征，所述目标图像特征中去除了人脸被遮挡的区域的图像特征；基于所述目标图像特征，对所述人脸图像中的人脸进行识别。在一种可能实现方式中，所述基于所述图像特征，确定对应的遮挡指示信息，包括：对所述图像特征进行卷积处理；对卷积处理后的图像特征进行分类，确定所述图像特征对应的遮挡指示信息。在一种可能实现方式中，所述根据所述图像特征和所述遮挡指示信息，获取目标图像特征，包括：基于所述遮挡指示信息，去除所述图像特征中所述人脸被遮挡区域的图像特征，得到目标图像特征。在一种可能实现方式中，所述根据所述图像特征和所述遮挡指示信息，获取目标图像特征，包括：将所述图像特征与所述遮挡指示信息相乘，得到所述目标图像特征。在一种可能实现方式中，所述对待识别的人脸图像进行特征提取，得到所述人脸图像的图像特征，包括：对所述待识别的人脸图像进行预处理；对预处理后的人脸图像进行特征提取，得到所述人脸图像的图像特征。在一种可能实现方式中，所述对所述待识别的人脸图像进行预处理，包括：对所述人脸图像进行人脸检测；基于人脸检测结果，对所述人脸图像进行裁剪，得到预处理后的人脸图像。在一种可能实现方式中，所述方法包括：将所述待识别的人脸图像输入图像处理模型中，由所述图像处理模型执行所述特征提取、确定遮挡指示信息、获取目标图像特征以及人脸识别过程，输出识别结果。一方面，提供了一种图像处理模型训练装置，所述装置包括：第一获取模块，用于基于图像处理模型，获取样本人脸图像的预测识别结果和遮挡指示信息，所述遮挡指示信息用于指示所述样本人脸图像的受人脸被遮挡区域影响的图像特征；第二获取模块，用于基于所述预测识别结果和所述样本人脸图像所标注的目标识别结果，获取识别误差；分类模块，用于基于所述样本人脸图像的遮挡指示信息，对所述样本人脸图像的遮挡模式进行分类，得到预测遮挡模式，所述遮挡模式用于指示所述样本人脸图像中人脸被遮挡区域的位置以及尺寸；第三获取模块，用于基于所述预测遮挡模式和所述样本人脸图像所标注的目标遮挡模式，获取分类误差；更新模块，用于根据所述识别误差和所述分类误差，对所述图像处理模型的模型参数进行更新。在一种可能实现方式中，所述分类模块用于：基于所述样本人脸图像的遮挡指示信息，获取所述样本人脸图像的遮挡区域信息；将所述样本人脸图像的遮挡区域信息与至少两个候选遮挡模式的遮挡区域信息进行匹配，将匹配度最大的候选遮挡模式确定为所述样本人脸图像的预测遮挡模式。在一种可能实现方式中，所述更新模块用于执行下述任一项：获取所述分类误差和所述分类误差的权重的乘积，获取所述乘积与所述识别误差之和作为目标误差，基于所述目标误差，对所述图像处理模型的模型参数进行更新；基于所述分类误差和所述识别误差各自的权重，对所述分类误差和所述识别误差进行加权，得到目标误差，基于所述目标误差，对所述图像处理模型的模型参数进行更新。在一种可能实现方式中，所述第一获取模块包括第一获取单元和识别单元；所述第一获取单元用于基于图像处理模型，获取样本人脸图像的图像特征以及所述图像特征对应的遮挡指示信息；所述识别单元用于基于所述图像特征和所述遮挡指示信息进行人脸识别，得到所述样本人脸图像的预测识别结果。在一种可能实现方式中，所述第一获取单元包括特征提取子单元和确定子单元；所述特征提取子单元用于基于图像处理模型，对样本人脸图像进行特征提取，得到所述样本人脸图像的图像特征；所述确定子单元用于基于所述样本人脸图像的图像特征，确定对应的遮挡指示信息。在一种可能实现方式中，所述特征提取子单元用于：对所述样本人脸图像的图像特征进行卷积处理；对卷积处理后的图像特征进行分类，确定所述图像特征对应的遮挡指示信息。在一种可能实现方式中，所述识别单元包括去除子单元和识别子单元；所述去除子单元用于基于所述遮挡指示信息，去除所述图像特征中所述人脸被遮挡区域的图像特征，得到目标图像特征；所述识别子单元用于根据所述目标图像特征，对所述样本人脸图像本文档来自技高网...

【技术保护点】
1.一种图像处理模型训练方法，其特征在于，所述方法包括：/n基于图像处理模型，获取样本人脸图像的预测识别结果和遮挡指示信息，所述遮挡指示信息用于指示所述样本人脸图像的受人脸被遮挡区域影响的图像特征；/n基于所述预测识别结果和所述样本人脸图像所标注的目标识别结果，获取识别误差；/n基于所述样本人脸图像的遮挡指示信息，对所述样本人脸图像的遮挡模式进行分类，得到预测遮挡模式，所述遮挡模式用于指示所述样本人脸图像中人脸被遮挡区域的位置以及尺寸；/n基于所述预测遮挡模式和所述样本人脸图像所标注的目标遮挡模式，获取分类误差；/n根据所述识别误差和所述分类误差，对所述图像处理模型的模型参数进行更新。/n

【技术特征摘要】
1.一种图像处理模型训练方法，其特征在于，所述方法包括：
基于图像处理模型，获取样本人脸图像的预测识别结果和遮挡指示信息，所述遮挡指示信息用于指示所述样本人脸图像的受人脸被遮挡区域影响的图像特征；
基于所述预测识别结果和所述样本人脸图像所标注的目标识别结果，获取识别误差；
基于所述样本人脸图像的遮挡指示信息，对所述样本人脸图像的遮挡模式进行分类，得到预测遮挡模式，所述遮挡模式用于指示所述样本人脸图像中人脸被遮挡区域的位置以及尺寸；
基于所述预测遮挡模式和所述样本人脸图像所标注的目标遮挡模式，获取分类误差；
根据所述识别误差和所述分类误差，对所述图像处理模型的模型参数进行更新。


2.根据权利要求1所述的方法，其特征在于，所述基于所述样本人脸图像的遮挡指示信息，对所述样本人脸图像的遮挡模式进行分类，得到预测遮挡模式，包括：
基于所述样本人脸图像的遮挡指示信息，获取所述样本人脸图像的遮挡区域信息；
将所述样本人脸图像的遮挡区域信息与至少两个候选遮挡模式的遮挡区域信息进行匹配，将匹配度最大的候选遮挡模式确定为所述样本人脸图像的预测遮挡模式。


3.根据权利要求1所述的方法，其特征在于，所述根据所述识别误差和所述分类误差，对所述图像处理模型的模型参数进行更新，包括下述任一项：
获取所述分类误差和所述分类误差的权重的乘积，获取所述乘积与所述识别误差之和作为目标误差，基于所述目标误差，对所述图像处理模型的模型参数进行更新；
基于所述分类误差和所述识别误差各自的权重，对所述分类误差和所述识别误差进行加权，得到目标误差，基于所述目标误差，对所述图像处理模型的模型参数进行更新。


4.根据权利要求1所述的方法，其特征在于，所述基于图像处理模型，获取样本人脸图像的预测识别结果和遮挡指示信息，包括：
基于图像处理模型，获取样本人脸图像的图像特征以及所述图像特征对应的遮挡指示信息；
基于所述图像特征和所述遮挡指示信息进行人脸识别，得到所述样本人脸图像的预测识别结果。


5.根据权利要求4所述的方法，其特征在于，所述基于图像处理模型，获取样本人脸图像的图像特征以及所述图像特征对应的遮挡指示信息，包括：
基于图像处理模型，对样本人脸图像进行特征提取，得到所述样本人脸图像的图像特征；
基于所述样本人脸图像的图像特征，确定对应的遮挡指示信息。


6.根据权利要求5所述的方法，其特征在于，所述基于所述样本人脸图像的图像特征，确定对应的遮挡指示信息，包括：
对所述样本人脸图像的图像特征进行卷积处理；
对卷积处理后的图像特征进行分类，确定所述图像特征对应的遮挡指示信息。


7.根据权利要求4所述的方法，其特征在于，所述基于所述图像特征和所述遮挡指示信息进行人脸识别，得到所述样本人脸图像的预测识别结果，包括：
基于所述遮挡指示信息，去除所述图像特征中所述人脸被遮挡区域的图像特征，得到目标图像特征；
根据所述目标图像特征，对所述样本人脸图像中的人脸进行识别，得到预测识别结果。


8.根据权利要求4所述的方法，其特征在于，所述基于图像处理模...

【专利技术属性】
技术研发人员：邱海波，龚迪洪，李志鋒，刘威，
申请(专利权)人：腾讯科技深圳有限公司，
类型：发明
国别省市：广东;44