[发明专利]白平衡校正方法、装置、设备和存储介质

权利要求书

1.一种白平衡校正方法，其特征在于，包括：

将待校正图像输入到预先训练的无色差点模型中，得到所述待校正图像的无色差点权重图；

根据所述无色差点权重图确定所述待校正图像的光照颜色参数；

根据所述光照颜色参数对所述待校正图像进行白平衡校正，得到校正图像。

2.根据权利要求1所述的方法，其特征在于，所述无色差点模型的训练过程包括：

获取训练样本集彩色图像；

通过预先构建的网络确定所述训练样本集彩色图像的样本无色差点权重图；

根据所述样本无色差点权重图确定所述训练样本集彩色图像的样本光照颜色参数；

根据所述样本光照颜色参数确定所述训练样本集彩色图像的样本校正图像；

根据所述样本校正图像的图像信息和所述样本无色差点权重图，对所述无色差点模型进行反向传播，以优化所述网络中的参数。

3.根据权利要求2所述的方法，其特征在于，所述训练样本集彩色图像不包括标签光照颜色参数；

相应的，根据所述样本校正图像的图像信息和所述样本无色差点权重图，对所述无色差点模型进行反向传播，以优化所述网络中的参数，包括：

构建无监督训练的损失函数；

基于所述损失函数，根据所述样本校正图像的图像信息和所述样本无色差点权重图，确定所述样本校正图像的误差值；

根据所述误差值对所述无色差点模型进行反向传播，以优化所述网络中的参数。

4.根据权利要求3所述的方法，其特征在于，通过如下公式构建无监督训练的损失函数：

其中，I表示待计算误差值的样本校正图像；i表示所述样本校正图像中预先划分的局部图像块的序号；n表示将所述样本校正图像划分为n个局部图像块；p_i表示第i个局部图像块的局部对比度；L_i表示第i个局部图像块的亮度值；t表示图像的三通道序号；SD_t表示局部图像块在t通道的局部方差；表示局部图像块在三通道上的局部均方差。

5.根据权利要求3所述的方法，其特征在于，基于所述损失函数，根据所述样本校正图像的图像信息和所述样本无色差点权重图，确定所述样本校正图像的误差值，包括：

根据所述样本校正图像的图像信息和所述样本无色差点权重图，通过公式t∈{R,G,B}确定校正权重图像；

基于所述损失函数和所述校正权重图像，确定所述样本校正图像的误差值；

其中，为所述校正权重图像中在(x,y)像素点处的t通道值，ω(x,y)为根据所述样本无色差点权重图确定的(x,y)像素点处的无色差点权重值，whiteⁱ(x,y)为所述样本校正图像中在(x,y)像素点处的t通道值，∑ω(x,y)为所述样本无色差点权重图的权重和。

6.根据权利要求2所述的方法，其特征在于，预先构建的网络中至少包括卷积层、激活函数层、池化层以及反卷积层，以使得输入训练样本集彩色图像和输出样本无色差点权重图具有相同分辨率。

7.根据权利要求1所述的方法，其特征在于，通过如下公式确定所述光照颜色参数e＝[r,g,b]：

其中，ω(x,y)为根据所述无色差点权重图确定的(x,y)像素点处的无色差点权重值，∑ω(x,y)为所述无色差点权重图的权重和，f^R(x,y)为所述待校正图像中在(x,y)像素点处的R通道值，f^G(x,y)为所述待校正图像中在(x,y)像素点处的G通道值，f^B(x,y)为所述待校正图像中在(x,y)像素点处的B通道值。