白平衡算法 python

以下是一个简单的白平衡算法 Python 实现示例：

import cv2
import numpy as np

def white_balance(image):
    # 转换为 LAB 颜色空间
    lab = cv2.cvtColor(image, cv2.COLOR_BGR2LAB)
    # 计算 a 和 b 通道的均值
    mean_a = np.mean(lab[:, :, 1])
    mean_b = np.mean(lab[:, :, 2])
    # 调整 a 和 b 通道的平均值
    lab[:, :, 1] = lab[:, :, 1] - ((mean_a - 128) * (lab[:, :, 0] / 255.0) * 1.1)
    lab[:, :, 2] = lab[:, :, 2] - ((mean_b - 128) * (lab[:, :, 0] / 255.0) * 1.1)
    # 转换回 RGB 颜色空间
    return cv2.cvtColor(lab, cv2.COLOR_LAB2BGR)

该算法首先将图像转换为 LAB 颜色空间，然后计算 a 和 b 通道的平均值。接下来，它将调整 a 和 b 通道的平均值，以便它们的值在整个图像中保持一致。最后，它将图像转换回 RGB 颜色空间，并返回结果。

使用示例：

image = cv2.imread('example.jpg')
result = white_balance(image)
cv2.imshow('Original', image)
cv2.imshow('White Balanced', result)
cv2.waitKey(0)