[发明专利]色彩平衡校正方法、装置、电子设备及存储介质

说明书

本申请提供了一种色彩平衡校正方法、装置、电子设备及存储介质。其中，该方法包括：确定待校正像素相应的第一色彩像素和第二色彩像素，第一色彩像素为与待校正像素在同一通道的像素，第二色彩像素为与待校正像素不在同一通道的同色像素；根据第一色彩像素计算待校正像素在当前位置的多个梯度值；根据第二色彩像素计算多个第二色彩像素的多个特征值；根据多个梯度值和多个特征值计算待校正像素的色彩平衡校正值，以通过色彩平衡校正值对待校正像素进行色彩平衡校正。本申请实施例通过根据多个梯度值和多个特征值对待校正像素的矫正值做出多种预测，并基于该多个梯度值和多个特征值得到的色彩平衡校正值，能够提高该色彩平衡校正值的准确性。

技术领域

本申请涉及图像处理领域，具体而言，涉及一种色彩平衡校正方法、装置、电子设备及存储介质。

背景技术

由于传感器内部存在串扰现象，即邻近像素的光通过某种方式泄露至当前像素，而影响了当前像素的像素值。受传感器的制造工艺的影响，上述光的泄露程度在横向和纵向是不同的，进而会造成同色像素对同样的光源，有不同的表现，因而在最后的成像中，图像上就会出现网格状的瑕疵，造成严重的画质问题。

目前，克服上述问题常用的做法是对串扰进行分析，校正，进而缓解色彩不平衡的问题。或利用图对图像的分析来人为判断是否存在色彩不平衡，从而进行相应的校正。但是串扰的成因十分复杂，涉及到镜头、传感器、光源等各种因素。因此，通过分析串扰的程度对色彩不平衡进行校正，是比较不准确的。而通过图像分析的手段进行色彩不平衡的校正，因不能判断图像上同色像素不同的表现是来自于色彩不平衡的问题，还是真的是纹理本身的问题。如果出现误判，则可能造成图像的解析力差等问题。

发明内容

有鉴于此，本申请实施例的目的在于提供一种色彩平衡校正方法、装置、电子设备及可读存储介质，能够提高该色彩平衡校正值的准确性，进而提高色彩不平衡校正的准确性。

第一方面，本申请实施例提供了一种色彩平衡校正方法，包括：确定待校正像素相应的第一色彩像素和第二色彩像素，所述第一色彩像素为与所述待校正像素在同一通道的像素，所述第二色彩像素为与所述待校正像素不在同一通道的同色像素；根据所述第一色彩像素计算所述待校正像素在当前位置的多个梯度值；根据所述第二色彩像素计算多个所述第二色彩像素的多个特征值；根据多个所述梯度值和多个所述特征值计算所述待校正像素的色彩平衡校正值，以通过所述色彩平衡校正值对所述待校正像素进行色彩平衡校正。

在上述实现过程中，通过根据第一色彩像素计算多个梯度值，第二色彩像素计算多个特征值，以通过多个梯度值和多个特征值对待校正像素的校正值做多种预测，并基于该多个梯度值和多个特征值得到的色彩平衡校正值，能够提高该色彩平衡校正值的准确性，进而提高色彩平衡矫正的准确性。

在一个实施例中，所述根据多个所述梯度值和多个所述特征值计算所述待校正像素的色彩平衡校正值，以通过所述色彩平衡校正值对所述待校正像素进行色彩平衡校正，包括：将所述多个所述梯度值和多个所述特征值进行排列组合后相加，以得到所述第二色彩像素的估计值；根据所述第二色彩像素的估计值和所述待校正像素的像素值计算所述待校正像素的色彩平衡校正值，以通过所述色彩平衡校正值对所述待校正像素进行色彩平衡校正。

在上述实现过程中，通过多个梯度值和多个特征值进行排列组合，得到多个第二色彩像素估计值，可以对该待校正像素的校正值做多种预测，提高了对该待校正像素的校正值预测的准确性。

在一个实施例中，所述根据所述第二色彩像素的估计值和所述待校正像素的像素值计算所述待校正像素的色彩平衡校正值，以通过所述色彩平衡校正值对所述待校正像素进行色彩平衡校正，包括：将所述第二色彩像素的估计值和所述待校正像素的像素值进行滤波处理，以得到所述第二色彩像素的最终估计值；将所述第二色彩像素的最终估计值和所述待校正像素的像素值进行加权平均处理，得到所述待校正像素的色彩平衡校正值，以通过所述色彩平衡校正值对所述待校正像素进行色彩平衡校正。