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

说明书

本发明实施例公开了一种色彩校正方法、装置、电子设备和存储介质，方法包括：根据环境亮度和图像传感器的各个区块输出的图像块数据的色温，确定各个图像块数据中的红外光占比；其中，所述图像传感器预先划分为至少两个区块；根据红外光占比消除对应图像块数据中各像素点的红外光；对消除红外光的图像块数据进行色彩校正。本发明实施例，基于环境亮度和传感器每个区块输出的图像块数据各自的色温，可精准确定每个图像块数据中各自的红外光占比，消除红外光后，进行色彩校正，由此解决设备处于混合色温场景时，图像传感器每块区域获取不同光源带来的影响，从而适应不同场景的色彩校正。

技术领域

本发明实施例涉及图像处理技术领域，尤其涉及一种色彩校正方法、装置、电子设备和存储介质。

背景技术

目前，由于RGB-IR图像传感器设备不需要IR-CUT红外截止滤镜，节省了硬件成本，另外该传感器的IR通道(传感器上覆盖了IR滤镜的像素)可以提高亮度信息，所以现在RGB-IR图像传感器越来越受到市场的欢迎，越来越多的应用到IPC设备中，但是由于RGB-IR的图像传感器R通道像素(传感器上覆盖了红色滤镜的像素)、G通道像素(传感器上覆盖了绿色滤镜的像素)、B通道像素(传感器上覆盖了蓝色滤镜的像素)接受到了多余的红外光，影响了AWB(自动白平衡)-CCM(色彩还原校正)，从而影响了图像实况的色彩还原，所以对于RGB-IR图像传感器的色彩还原校正越来越重要。

而现有的色彩校正方法，一般先在不同光照条件下设备在标准650nm截止滤波器和标准850nm截止滤波器拍摄色卡数据，然后通过对比没有红外截止滤波器拍摄色卡的平均的RGBIR值，进而利用最小二乘法拟合得到每个色温环境下的红外光占比，然后根据设备所处的实际环境对整幅图像的每个像素点按照当前色温对比预先标定表的数据查找该色温对应的红外光占比、消除红外光的影响，进行色彩还原校正。

然而这种方法仍存在一定的不足：没有考虑同一色温下、不同的曝光时间下图像传感器的各个色彩通道对光源的响应，使得拟合得到的红外光占比计算不准确，而且默认把红外光敏传感器感受到的色温当作整副图像所有像素点的当前色温，进而确定红外光占比，使得计算精准性低，不适用于混合色温场景和一些更复杂场景。

发明内容

本发明实施例提供一种色彩校正方法、装置、电子设备和存储介质，以达到使RGB-IR图像传感器设备在不同场景的精准进行色彩还原的目的。

第一方面，本发明实施例提供了一种色彩校正方法，所述方法包括：

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

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

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

第二方面，本发明实施例提供了一种色彩校正装置，所述装置包括：

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

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

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

第三方面，本发明实施例还提供了一种电子设备，包括：

一个或多个处理器；

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

当所述一个或多个程序被所述一个或多个处理器执行，使得所述一个或多个处理器实现如本发明任一实施例所述的色彩校正方法。