[发明专利]白平衡校正方法、装置、设备和存储介质

说明书

本发明实施例公开了一种白平衡校正方法、装置、设备和存储介质。该白平衡校正方法包括：将待校正图像输入到预先训练的无色差点模型中，得到所述待校正图像的无色差点权重图；根据所述无色差点权重图确定所述待校正图像的光照颜色参数；根据所述光照颜色参数对所述待校正图像进行白平衡校正，得到校正图像。本发明实施例基于预先训练的无色差点模型确定待校正图像的无色差点权重图，根据无色差点权重图得到待校正图像的光照颜色，最后根据光照颜色进行白平衡校正，实现对待校正图像的白平衡校正。本发明实施例的白平衡校正方法不对待校正图像的场景进行限制，能够提高白平衡校正的自适应性，提高图像校正结果的准确度。

技术领域

本发明实施例涉及互联网技术领域，尤其涉及一种白平衡校正方法、装置、 设备和存储介质。

背景技术

自然界中，物体所呈现的颜色会随着所在环境光源的变化而变化，对于人 类而言，即使光源有所改变但是眼睛对于色彩的感知并没有受到很大影响。这 是由于大脑对物体的颜色具有一定的先验知识，能从某些变化的光源下察觉物 体固有的颜色。而摄像机本身不具备人眼的这种特殊功能，其在不同光源下拍 摄图像时会受到光源颜色的影响，因此需要对摄像机获取的图像做白平衡校正 处理。

目前，常用的白平衡校正算法是基于统计的方法。即在某种假设下，通过 统计场景图像自身的颜色特征来估计出环境光源。例如常见的灰度世界法、完 美反射法等，这类方法原理简单、易于实现，但由于基于的假设性太强，导致 使用场景受限。基于统计的方法普遍存在假设性较强，自适应能力差的缺点。

发明内容

本发明实施例提供一种白平衡校正方法、装置、设备和存储介质，以提高 白平衡校正的自适应性，进而提高校正结果的准确性。

第一方面，本发明实施例提供了一种白平衡校正方法，包括：

将待校正图像输入到预先训练的无色差点模型中，得到所述待校正图像的 无色差点权重图；

根据所述无色差点权重图确定所述待校正图像的光照颜色参数；

根据所述光照颜色参数对所述待校正图像进行白平衡校正，得到校正图像。

第二方面，本发明实施例还提供了一种白平衡校正装置，包括：

权重图确定模块，用于将待校正图像输入到预先训练的无色差点模型中， 得到所述待校正图像的无色差点权重图；

光照颜色确定模块，用于根据所述无色差点权重图确定所述待校正图像的 光照颜色参数；

白平衡校正模块，用于根据所述光照颜色参数对所述待校正图像进行白平 衡校正，得到校正图像。

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

一个或多个处理器；

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

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

第四方面，本发明实施例还提供了一种计算机可读存储介质，其上存储有 计算机程序，该程序被处理器执行时实现如本发明任一实施例所述的白平衡校 正方法。

本发明实施例基于预先训练的无色差点模型确定待校正图像的无色差点权 重图，根据无色差点权重图得到待校正图像的光照颜色，最后根据光照颜色进 行白平衡校正，实现对待校正图像的白平衡校正。本发明实施例的白平衡校正 方法不对待校正图像的场景进行限制，能够提高白平衡校正的自适应性，提高 图像校正结果的准确度。

附图说明

图1是本发明实施例一中的白平衡校正方法的流程图；

图2A是本发明实施例二中的无色差点模型训练过程的流程图；