【技术实现步骤摘要】
本申请涉及终端，尤其涉及一种模型训练方法、图像处理方法及电子设备。
技术介绍
1、在终端设备中通常安装有用于实现图像拍摄功能的应用程序，这些应用程序需要对摄像头所采集的原始图像进行图像处理，才能够得到可以显示的输出图像。
2、例如可以在应用程序中部署图像处理模型，之后通过图像处理模型对原始图像进行图像处理，从而得到输出图像。在理想情况下，图像处理模型对原始图像所进行的图像处理，不会改变原始图像的色调信息。因为原始图像的色调信息一般都是比较准确的，因此可以保证最终的输出图像的色调表现正常。
3、然而，在模型部署过程中，出于性能考虑，可能会损失一部分模型精度，因此可能会导致最终的输出图像存在偏色问题。
技术实现思路
1、本申请实施例提供一种模型训练方法、图像处理方法及电子设备，应用于终端
通过在图像处理模型的输入数据中增加至少一个目标颜色通道的色调偏移系数，以使得图像处理模型可以在图像处理过程中根据色调偏移系数进行相应程度的色调偏移，从而针对性的修正模型处理图像偏色的问题。
2、第一方面，本申请实施例提出一种模型训练方法。该方法包括：
3、确定至少一个目标颜色通道的样本偏移系数，所述样本偏移系数用于指示对所述目标颜色通道对应的色调参数进行偏移；
4、将所述样本图像输入至图像处理模型，以得到所述图像处理模型输出的第一处理图像，并确定所述第一处理图像的色调参数；
5、根据所述样本偏移系数，对所述第一处理图像的色
6、将所述样本图像以及所述至少一个目标颜色通道的样本偏移系数输入至图像处理模型，以得到所述图像处理模型输出的第二处理图像，并确定所述第二处理图像的实际色调参数；
7、根据所述参考色调参数以及所述实际色调参数，更新所述图像处理模型的模型参数。
8、这样，可以使得图像处理模型学习到如下能力：在图像处理过程中，根据输入的样本偏移参数对相应的颜色通道的色调参数进行偏移处理。当图像处理模型具备这样的能力时，就可以在模型应用阶段，根据实际的偏色情况设置色调偏移系数，以使得图像处理模型根据色调偏移系数对图像进行色调偏移，从而实现对偏色的纠正。
9、在一种可能的实现方式中，所述色调参数为自动白平衡awb增益，所述awb增益中包括至少一个颜色通道的增益参数；
10、所述确定至少一个目标颜色通道的样本偏移系数，包括：
11、针对任一个所述目标颜色通道，在所述样本图像的至少一个颜色通道的增益参数中，将所述目标颜色通道的增益参数确定为第一初始增益参数；
12、对所述第一初始增益参数进行数值调整处理，得到所述目标颜色通道的第一调整增益参数；
13、将所述第一初始增益参数和所述第一调整增益参数的比值，确定为所述目标颜色通道的样本偏移系数。
14、在一种可能的实现方式中，所述对所述第一初始增益参数进行数值调整处理，得到所述目标颜色通道的第一调整增益参数，包括：
15、在预设数值范围内生成第一数值，根据所述第一数值对所述第一初始增益参数进行所述数值调整处理，得到所述目标颜色通道的第一调整增益参数；
16、其中，所述数值调整处理为在调整类型中随机选择的，或者，所述数值调整处理为在所述调整类型中预先确定的，所述调整类型包括：增大数值、减小数值以及保持数值不变。
17、在这种实现方式中，可以在样本图像的基础上，对色调参数进行数值调整，从而构造虚拟的偏色场景，以确定样本偏移系数，这种实现方式可以保证确定的样本偏移系数的合理性和实用性。
18、在一种可能的实现方式中，所述根据所述样本偏移系数，对所述第一处理图像的色调参数进行偏移处理，得到参考色调参数，包括：
19、针对任一个所述目标颜色通道，在所述第一处理图像的至少一个颜色通道的增益参数中，将所述目标颜色通道的增益参数确定为第二初始增益参数；
20、根据所述目标颜色通道的样本偏移系数与所述第二初始增益参数的乘积，确定所述目标颜色通道的第二调整增益参数；
21、将所述第一处理图像的至少一个颜色通道的增益参数中，所述目标颜色通道的增益参数替换为所述第二调整增益参数，以得到所述参考色调参数。
22、在这种实现方式中，可以有效的构造出标注数据，也就是期望的模型输出的相关数据，从而可以对模型进行正确的反馈，以有效的优化模型。
23、第二方面，本申请实施例提出一种图像处理方法。该方法包括：
24、获取摄像头采集的原始图像，以及获取至少一个目标颜色通道的色调偏移系数，其中，所述色调偏移系数用于指示对所述目标颜色通道对应的色调参数进行偏移处理；
25、将所述原始图像以及所述至少一个目标颜色通道的色调偏移系数输入至图像处理模型，以得到所述图像处理模型输出的目标图像，所述图像处理模型为根据第一方面所述的方法训练得到的。
26、这样，通过在输入数据中增加色调偏移系数，以控制图像出图的颜色倾向，具体的过控制图像处理模型参照色调偏移系数，在图像处理过程中对图像的色调参数进行相应程度的偏移处理，从而实现对图像处理模型的偏色处理的补偿，以保证图像处理模型最终输出的目标图像能够正确表现实际环境中的色调。
27、在一种可能的实现方式中，所述获取至少一个目标颜色通道的色调偏移系数，包括：
28、将测试输入图像输入至图像处理模型，以得到所述图像处理模型输出的测试输出图像；
29、根据所述测试输入图像的色调参数以及所述测试输出图像的色调参数，确定所述至少一个目标颜色通道的色调偏移系数。在一种实现方式中，例如针对任一个目标颜色通道，可以将测试输入图像中目标颜色通道的色调参数与测试输出图像中目标颜色通道的色调参数的比值，确定为该目标颜色通道的色调偏移系数。
30、在这种实现方式中，通过对模型进行测试，以根据模型的输入以及输出确定目标颜色通道的色调偏移系数，从而可以保证色调偏移系数的正确性。
31、在一种可能的实现方式中，所述图像处理模型中的处理单元包括如下中的至少一种：
32、黑电平校正单元、自动白平衡单元、集成处理单元、局部颜色纠偏单元以及镜头阴影校正单元，其中，所述集成处理单元用于实现如下功能中的至少一种：降噪处理、去马赛克处理以及高动态范围处理。
33、其中，所述局部颜色纠偏单元用于根据补偿矩阵对集成处理单元输出的中间图像进行偏色补偿。
34、在一种可能的实现方式中，所述方法还包括：
35、在所述图像处理模型的处理过程中，获取所述集成处理单元输出的中间图像；
36、在所述原始图像和所述中间图像中均划分n个图像区域，其中所述原始图像中的n个图像区域和所述中间图像中的n个图像区域一一对应，所述n为大于或等于1的整数；
37、针对任一个所述图像区域，本文档来自技高网...
【技术保护点】
1.一种模型训练方法，其特征在于，包括：
2.根据权利要求1所述的方法，其特征在于，所述色调参数为自动白平衡AWB增益，所述AWB增益中包括至少一个颜色通道的增益参数；
3.根据权利要求2所述的方法，其特征在于，所述对所述第一初始增益参数进行数值调整处理，得到所述目标颜色通道的第一调整增益参数，包括：
4.根据权利要求2或3所述的方法，其特征在于，所述根据所述样本偏移系数，对所述第一处理图像的色调参数进行偏移处理，得到参考色调参数，包括：
5.一种图像处理方法，其特征在于，包括：
6.根据权利要求5所述的方法，其特征在于，所述获取至少一个目标颜色通道的色调偏移系数，包括：
7.根据权利要求5或6所述的方法，其特征在于，所述图像处理模型中的处理单元包括如下中的至少一种：
8.根据权利要求7所述的方法，其特征在于，所述方法还包括：
9.根据权利要求8所述的方法，其特征在于，所述根据各所述图像区域各自对应的至少一个目标颜色通道的区域补偿值，确定所述至少一个目标颜色通道的第一补偿矩阵，包括：<...
【技术特征摘要】
1.一种模型训练方法，其特征在于，包括：
2.根据权利要求1所述的方法，其特征在于，所述色调参数为自动白平衡awb增益，所述awb增益中包括至少一个颜色通道的增益参数；
3.根据权利要求2所述的方法，其特征在于，所述对所述第一初始增益参数进行数值调整处理，得到所述目标颜色通道的第一调整增益参数，包括：
4.根据权利要求2或3所述的方法，其特征在于，所述根据所述样本偏移系数，对所述第一处理图像的色调参数进行偏移处理，得到参考色调参数，包括：
5.一种图像处理方法，其特征在于，包括：
6.根据权利要求5所述的方法，其特征在于，所述获取至少一个目标颜色通道的色调偏移系数，包括：
7.根据权利要求5或6所述的方法，其特征在于，所述图像处理模型中的处理单元包括如下中的至少一种：
8.根据权利要求7所述的方法，其特征在于，所述方法还包括：
9.根据权利要求8所述的方法，其特征在于，所述根据各所述图像区域各自对应的至少一个目标颜色通道的区域补偿值，确定所述至少一个目标颜色通道的第一补偿矩阵，包括：
10.根据权利要求9所述的方法，其特征在于，所述将各所述图像区域中的各所...
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。