[发明专利]图像颜色恢复方法和装置、摄像装置、存储介质

说明书

图像颜色恢复方法及装置，摄像装置，以及存储介质。所述方法包括：获取原始图像；确定所述原始图像每个像素对应的C、M、Y、W四个颜色通道值；基于所述C、M、Y、W四个颜色通道值，计算所述原始图像的亮度分量以及每个像素对应的R、G、B三个颜色通道值，获得第一RGB图像；计算所述第一RGB图像的亮度分量和颜色分量；基于所述原始图像的亮度分量、所述第一RGB图像的亮度分量和颜色分量，计算所述第一RGB图像对应的YUV图像的亮度分量和颜色分量，以获得所述YUV图像；以及将所述YUV图像转换为第二RGB图像，作为颜色恢复后的图像。基于本发明，可以在不使用IRCUT的条件下恢复图像的颜色，保留近红外光在透雾、低照度时降噪和提高图像亮度等方面的优点。

技术领域

本发明涉及图像处理领域，具体地涉及一种图像颜色恢复方法和装置、摄像装置、存储介质。

背景技术

目前常见的图像传感器，包括电荷耦合器件(Charge Coupled Devices，CCD)图像传感器和互补金属氧化物半导体(Complementary Metal Oxide Semiconductor，CMOS)图像传感器，均为硅基图像传感器，其在可见光之外的近红外波段仍能感光。近红外光进入图像传感器后，将冲淡图像的颜色，例如绿叶植物会强烈反射近红外光，导致拍摄的草地、树叶呈现雪景般的白色。通常要在彩色图像传感器前面安装红外截止滤光片(Infrared Cut-off filter，IRCUT)，以滤除近红外光、保留可见光，以恢复出与人类视觉相接近的色彩。

然而，安装IRCUT存在以下几个方面的缺点。首先，近红外光在透雾方面有重要作用，安装IRCUT之后，将失去近红外光透雾的功能；其次，在曝光不足的场景下，例如室内低照度场景和夜晚，近红外光有利于降低噪声、提高图像亮度。安装IRCUT后将失去近红外光的这些作用。此外，安装IRCUT还会增加设备成本。

现有技术中的一些颜色校正方法可以在不使用IRCUT的情况下恢复图像的颜色，但其无法保留近红外光在透雾、低照度时降噪和提高图像亮度等方面的优点。

发明内容

本发明解决的技术问题是如何在不使用IRCUT的条件下恢复图像的颜色，同时保留近红外光在透雾、低照度时降噪和提高图像亮度等方面的优点。

为解决上述技术问题，本发明实施例提供一种图像颜色恢复方法，包括：获取原始图像；确定所述原始图像每个像素对应的C、M、Y、W四个颜色通道值；基于所述C、M、Y、W四个颜色通道值，计算所述原始图像的亮度分量以及每个像素对应的R、G、B三个颜色通道值，获得第一RGB图像；计算所述第一RGB图像的亮度分量和颜色分量；基于所述原始图像的亮度分量、所述第一RGB图像的亮度分量和颜色分量，计算所述第一RGB图像对应的YUV图像的亮度分量和颜色分量，以获得所述YUV图像；以及将所述YUV图像转换为第二RGB图像，作为颜色恢复后的图像。

可选地，所述原始图像基于具有CMYW滤色器阵列的图像传感器获得。

可选地，确定所述原始图像每个像素对应的C、M、Y、W四个颜色通道值包括：基于去马赛克处理，确定所述原始图像每个像素对应的C、M、Y、W四个颜色通道值。

可选地，所述原始图像每个像素对应的R、G、B三个颜色通道值基于公式(1)而计算：

其中，C、M、Y、W分别是所述原始图像的C、M、Y、W四个颜色通道值，R、G、B分别是对应的R、G、B三个颜色通道值，k1、k2、k3是修正系数。

可选地，所述原始图像的亮度分量基于公式(2)而计算：

Y0＝(C+M+Y+W)/4 (2)，

其中，Y0是所述原始图像的亮度分量，C、M、Y、W分别是所述原始图像的C、M、Y、W四个颜色通道值。