图像白平衡调整方法技术

本申请涉及图像处理技术领域，提供了一种图像白平衡调整方法

【技术实现步骤摘要】
图像白平衡调整方法、装置、电子设备及存储介质


[0001]本申请涉及图像处理
，尤其是涉及一种图像白平衡调整方法
、
装置
、
电子设备及存储介质
。

技术介绍

[0002]白平衡用于确保图像能够适当地反映拍摄图像的真实世界场景中的实际白色
。
为了减少照射光源的色温对物体颜色的影响，需要对图像进行白平衡处理，使得图像中的物体呈现出实际的颜色
。
当摄影或拍摄的环境中存在不同颜色的光源时，白平衡处理可以消除图像中的色偏，使图像中的白色看起来是真实的中性白色，以便在不同的光源条件下呈现准确的颜色
。
[0003]现有技术通常使用全局白平衡技术对图像进行校正，但是当图像的拍摄环境复杂时，全局自动白平衡技术的色彩校正准确性较低
。
这是因为全局自动白平衡技术通常基于整个图像的平均颜色来进行校正，无法针对不同的光源或背景环境进行精确调整
。

技术实现思路

[0004]有鉴于此，本申请提供一种图像白平衡调整方法
、
装置
、
电子设备及存储介质，以解决全局自动白平衡技术的色彩校正准确性较低的技术问题
。
[0005]本申请的第一方面提供一种图像白平衡调整方法，所述方法包括：获取彩色图像的整体特征向量及整体色温值；将所述彩色图像分为多个彩色图像块，并获取每个所述彩色图像块的块特征向量及块色温值；将所述整体特征向量及所述整体色温值进行正交得到第一正交向量；将每个所述块特征向量与对应的所述块色温值进行正交得到第二正交向量；将所述彩色图像
、
所述第一正交向量及多个所述第二正交向量输入白平衡调整模型，得到候选白平衡图像；基于预设标准色温阈值对所述候选白平衡图像进行调整，得到目标白平衡图像
。
[0006]在一种可能的实施方式中，所述将所述彩色图像分为多个彩色图像块包括：对所述彩色图像进行边缘检测，得到多个物体边缘；基于所述多个物体边缘将所述彩色图像分割为多个彩色图像块
。
[0007]在一种可能的实施方式中，所述获取彩色图像的整体特征向量及整体色温值包括：获取所述彩色图像在
RGB
颜色空间的通道值；将所述彩色图像由
RGB
颜色空间转换为指定颜色空间；获取所述彩色图像在所述指定颜色空间中的色度值
、
亮度值和颜色分布直方图；将所述通道值
、
所述亮度值以及所述颜色分布直方图作为所述整体特征向量；基于所述色度值及所述指定颜色空间对应的色度
‑
色温关系模型计算所述彩色图
像的整体色温值
。
[0008]在一种可能的实施方式中，所述白平衡调整模型通过如下方式训练得到：基于卷积神经网络构建初始白平衡调整模型；获取每个所述彩色样本图像对应的参考白平衡图像；基于所述彩色样本图像的整体特征向量及整体色温值得到第一正交样本向量；基于所述彩色样本图像的块特征向量及对应的块色温值得到第二正交样本向量；将所述彩色样本图像
、
所述参考白平衡图像以及所述第一正交样本向量及对应的所述第二正交样本向量输入所述初始白平衡调整模型中进行训练；基于最小化预设损失函数对所述初始白平衡调整模型的模型参数进行优化，得到所述白平衡调整模型
。
[0009]在一种可能的实施方式中，所述获取每个所述彩色样本图像对应的参考白平衡图像包括：将所述彩色样本图像从
RGB
颜色空间转换到
Lab
颜色空间；对所述彩色样本图像进行去噪处理，得到去噪图像；获取所述去噪图像的平均亮度分量值，根据所述平均亮度分量值计算得到亮度调整因子，并基于所述亮度调整因子对所述去噪图像进行亮度调整；获取亮度调整后的去噪图像的平均
a
色度分量值及平均
b
色度分量值，根据所述平均
a
色度分量值计算得到
a
色度分量缩放因子，根据所述平均
b
色度分量值计算得到
b
色度分量缩放因子，并基于所述
a
色度分量缩放因子及所述
b
色度分量缩放因子对所述亮度调整后的去噪图像进行色度调整，得到所述参考白平衡图像
。
[0010]在一种可能的实施方式中，所述基于预设标准色温阈值对所述候选白平衡图像进行调整，得到目标白平衡图像包括：获取所述候选白平衡图像的候选色温值；将所述候选色温值与所述预设标准色温阈值进行比较；当所述候选色温值高于所述预设标准色温阈值时，使用预设第一调整系数对所述候选白平衡图像进行调整，得到目标白平衡图像；当所述候选色温值低于所述预设标准色温阈值时，使用预设第二调整系数对所述候选白平衡图像进行调整，得到目标白平衡图像；其中，所述预设第一调整系数小于1，所述预设第二调整系数大于
1。
[0011]在一种可能的实施方式中，在所述将所述候选色温值与所述预设标准色温阈值进行比较之前，所述方法还包括：判断所述候选色温值是否在预设标准色温阈值范围内；当确定所述候选色温值在所述预设标准色温阈值范围内时，执行所述基于预设标准色温阈值对所述候选白平衡图像进行调整，得到目标白平衡图像；当确定所述候选色温值不在所述预设标准色温阈值范围内时，获取所述候选白平衡图像的第一正交向量及多个第二正交向量，并将所述候选白平衡图像以及对应的所述第一正交向量及对应的多个所述第二正交向量输入所述白平衡调整模型，得到新的候选白平衡图像
。
[0012]本申请的第二方面提供一种图像白平衡调整装置，所述系统包括：
整体获取模块，用于获取彩色图像的整体特征向量及整体色温值；局部获取模块，用于将所述彩色图像分为多个彩色图像块，并获取每个所述彩色图像块的块特征向量及块色温值；第一正交模块，用于将所述整体特征向量及所述整体色温值进行正交得到第一正交向量；第二正交模块，用于将每个所述块特征向量与对应的所述块色温值进行正交得到第二正交向量；模型计算模块，用于将所述彩色图像
、
所述第一正交向量及多个所述第二正交向量输入白平衡调整模型，得到候选白平衡图像；输出调整模块，用于基于预设标准色温阈值对所述候选白平衡图像进行调整，得到目标白平衡图像
。
[0013]本申请的第三方面提供一种电子设备，包括存储器
、
处理器以及存储在所述存储器上并可在所述处理器上运行的计算机程序，所述处理器执行所述计算机程序时实现所述的图像白平衡调整方法的步骤
。
[0014]本申请的第四方面提供一种计算机可读存储介质，其上存储有计算机程序，所述计算机程序被处理器执行时实现所述的图像白平衡调整方法的步骤
。
[0015]本申请实施例提供的图像白平衡调整方法
、
装置
、
电子设备及存储介质，通过提取整体和局部的图像特征向量和色温值，本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.
一种白平衡调整方法图像白平衡调整方法，其特征在于，所述方法包括：获取彩色图像的整体特征向量及整体色温值；将所述彩色图像分为多个彩色图像块，并获取每个所述彩色图像块的块特征向量及块色温值；将所述整体特征向量及所述整体色温值进行正交得到第一正交向量；将每个所述块特征向量与对应的所述块色温值进行正交得到第二正交向量；将所述彩色图像
、
所述第一正交向量及多个所述第二正交向量输入白平衡调整模型，得到候选白平衡图像；基于预设标准色温阈值对所述候选白平衡图像进行调整，得到目标白平衡图像
。2.
根据权利要求1所述的图像白平衡调整方法，其特征在于，所述将所述彩色图像分为多个彩色图像块包括：对所述彩色图像进行边缘检测，得到多个物体边缘；基于所述多个物体边缘将所述彩色图像分割为多个彩色图像块
。3.
根据权利要求1所述的图像白平衡调整方法，其特征在于，所述获取彩色图像的整体特征向量及整体色温值包括：获取所述彩色图像在
RGB
颜色空间的通道值；将所述彩色图像由
RGB
颜色空间转换为指定颜色空间；获取所述彩色图像在所述指定颜色空间中的色度值
、
亮度值和颜色分布直方图；将所述通道值
、
所述亮度值以及所述颜色分布直方图作为所述整体特征向量；基于所述色度值及所述指定颜色空间对应的色度
‑
色温关系模型计算所述彩色图像的整体色温值
。4.
根据权利要求2所述的图像白平衡调整方法，其特征在于，所述白平衡调整模型通过如下方式训练得到：基于卷积神经网络构建初始白平衡调整模型；获取每个所述彩色样本图像对应的参考白平衡图像；基于所述彩色样本图像的整体特征向量及整体色温值得到第一正交样本向量；基于所述彩色样本图像的块特征向量及对应的块色温值得到第二正交样本向量；将所述彩色样本图像
、
所述参考白平衡图像以及所述第一正交样本向量及对应的所述第二正交样本向量输入所述初始白平衡调整模型中进行训练；基于最小化预设损失函数对所述初始白平衡调整模型的模型参数进行优化，得到所述白平衡调整模型
。5.
根据权利要求4所述的图像白平衡调整方法，其特征在于，所述获取每个所述彩色样本图像对应的参考白平衡图像包括：将所述彩色样本图像从
RGB
颜色空间转换到
Lab
颜色空间；对所述彩色样本图像进行去噪处理，得到去噪图像；获取所述去噪图像的平均亮度分量值，根据所述平均亮度分量值计算得到亮度调整因子，并基于所述亮度调整因子对所述去噪图像进行亮度调整；获取亮度调整后的去噪图像的平均
a
色度分量值及平均
b
色度分量值，根据所述平均
a
色度分量值计算得到
...

【专利技术属性】
技术研发人员：叶剑锋，苏星华，甄茂华，
申请(专利权)人：众显科技深圳有限公司，
类型：发明
国别省市：