[发明专利]一种色彩校正方法、装置、电子设备和存储介质

权利要求书

1.一种色彩校正方法，其特征在于，包括：

根据环境亮度和图像传感器的各个区块输出的图像块数据的色温，确定各个图像块数据中的红外光占比；其中，所述图像传感器预先划分为至少两个区块；

根据红外光占比消除对应图像块数据中各像素点的红外光；

对消除红外光的图像块数据进行色彩校正；

其中，根据环境亮度和图像传感器的各个区块输出的图像块数据的色温，确定各个图像块数据中的红外光占比之前，还包括：

根据所述图像传感器的各个区块输出的图像块数据在白区内的白点落位，计算该区块输出的图像块数据的通道增益；

基于该区块输出的图像块数据的通道增益和色温曲线函数，得到该区块输出的图像块数据的色温。

2.根据权利要求1所述的方法，其特征在于，对消除红外光的图像块数据进行色彩校正，包括：

根据已消除红外光的图像块数据的色温所对应的色彩校正矩阵，对该图像块数据进行色彩校正。

3.根据权利要求1所述的方法，其特征在于，所述图像传感器为RGB-IR图像传感器；

相应的，根据环境亮度和图像传感器的各个区块输出的图像块数据的色温，确定各个图像块数据中的红外光占比之前，还包括：

在获取的RGB-IR格式的图像数据中，确定每个基础单元中的IR通道数值，并将该IR通道数值作为该基础单元的红外光含量；其中RGB-IR格式的图像数据中包括多个基础单元，每个基础单元包括R、G、B、IR四个通道数据；

消除每个基础单元中的红外光，并对各个基础单元进行白平衡校正和色彩校正。

4.根据权利要求3所述的方法，其特征在于，消除每个基础单元中的红外光后，还包括：

获取消除红外光的输出数据，通过插值算法确定每个像素点的R、G、B值。

5.一种色彩校正装置，其特征在于，包括：

红外占比确定模块，根据环境亮度和图像传感器的各个区块输出的图像块数据的色温，确定各个图像块数据中的红外光占比；其中，所述图像传感器预先划分为至少两个区块；

红外消除模块，用于根据红外光占比消除对应图像块数据中各像素点的红外光；

色彩校正模块，用于对消除红外光的图像块数据进行色彩校正；

其中，获取模块包括：

通道增益计算模块，用于根据所述图像传感器的各个区块输出的图像块数据在白区内的白点落位，计算该区块输出的图像块数据的通道增益；

色温计算模块，用于基于该区块输出的图像块数据的通道增益和色温曲线函数，得到该区块输出的图像块数据的色温。

6.根据权利要求5所述的装置，其特征在于，所述色彩校正模块具体用于：

根据已消除红外光的图像块数据的色温所对应的色彩校正矩阵，对该图像块数据进行色彩校正。

7.一种电子设备，其特征在于，包括：

一个或多个处理器；

存储装置，用于存储一个或多个程序，

当所述一个或多个程序被所述一个或多个处理器执行，使得所述一个或多个处理器实现如权利要求1-4中任一所述的色彩校正方法。

8.一种计算机可读存储介质，其上存储有计算机程序，其特征在于，该程序被处理器执行时实现如权利要求1-4中任一所述的色彩校正方法。