[发明专利]图像处理模型的训练方法和装置、白平衡处理方法和装置

说明书

本申请公开了一种图像处理模型的训练方法和装置、白平衡处理方法和装置，其中，图像处理模型的训练方法包括：获取第一训练集图像；对第一训练集图像进行偏色处理，得到偏色后的第二训练集图像；通过第一训练集图像和第二训练集图像对待训练神经网络进行训练，得到用于白平衡处理的图像处理模型。

技术领域

本申请属于图像处理技术领域，具体涉及一种图像处理模型的训练方法和装置、白平衡处理方法和装置。

背景技术

在相关技术中，对于相机等图像获取设备，由于硬件传感器的局限，传感器输出的图像与人眼观看的图像会存在一定的差别，导致拍摄图像中物体颜色与人眼识别的被摄物体的实际颜色不同。

一般来说，需要对拍摄的照片源片进行白平衡处理，目前基于深度神经网络的白平衡方法基于监督和半监督方法，而该方法在模型训练过程中，需要大量包含真实光源注释的训练集数据，而真实光源注释需要消耗大量的时间，造成训练集获取困难，最终降低训练所得模型的准确性。

因此，目前亟需一种能够不依赖真实光源注释的训练集数据的训练方法。

发明内容

本申请实施例的目的是提供一种图像处理模型的训练方法和装置、图像的白平衡处理方法和装置，能够不依赖真实光源注释的训练集数据来对图像处理模型进行高效训练。

为了解决上述技术问题，本申请是这样实现的：

第一方面，本申请实施例提供了一种图像处理模型的训练方法，包括：

获取第一训练集图像；

对第一训练集图像进行偏色处理，得到偏色后的第二训练集图像；

通过第一训练集图像和第二训练集图像对待训练神经网络进行训练，得到用于白平衡处理的图像处理模型。

第二方面，本申请实施例提供了一种图像处理模型的训练装置，包括：

第一获取单元，用于获取第一训练集图像；

第一处理单元，用于对第一训练集图像进行偏色处理，得到偏色后的第二训练集图像；

训练单元，通过第一训练集图像和第二训练集图像对待训练神经网络进行训练，得到用于白平衡处理的图像处理模型。

第三方面，本申请实施例提供了一种图像的白平衡处理方法，包括：

获取根据第一方面中提供的图像处理模型的训练方法训练得到的图像处理模型；

通过图像处理模型对待处理图像进行白平衡处理，得到白平衡处理后的目标图像。

第四方面，本申请实施例提供了一种图像的白平衡处理装置，包括：

第二获取单元，用于获取根据第一方面中提供的图像处理模型的训练方法训练得到的图像处理模型；

第二处理单元，用于通过图像处理模型对待处理图像进行白平衡处理，得到白平衡处理后的目标图像。

第五方面，本申请实施例提供了一种电子设备，包括处理器，存储器及存储在存储器上并可在处理器上运行的程序或指令，程序或指令被处理器执行时实现如第一方面中提供的图像处理模型的训练方法；和/或

如第三方面中提供的图像的白平衡处理方法。

第六方面，本申请实施例提供了一种可读存储介质，可读存储介质上存储程序或指令，程序或指令被处理器执行时实现如第一方面中提供的图像处理模型的训练方法；和/或

如第三方面中提供的图像的白平衡处理方法。