python opencv correctColorBalance 示例

在Python中，OpenCV提供了一个名为`cv2.correctColorBalance`的方法用于调整图像的颜色平衡，这有助于校正图像的整体色调。下面是一个简单的示例，展示如何使用这个功能：

import cv2
import numpy as np

# 加载图像
img = cv2.imread('your_image_path.jpg', cv2.IMREAD_COLOR)

# 获取图像的大小
height, width, _ = img.shape

# 创建一个3x3的矩阵，分别代表红、绿、蓝三个通道的增益
gain = np.array([[0.6, 0.75, 0.8],
                 [0.9, 1.0, 1.1],
                 [0.7, 0.8, 1.0]])

# 应用color balance
balanced_img = cv2.convertScaleAbs(img, gain)

# 显示原图和处理后的图像
cv2.imshow("Original Image", img)
cv2.imshow("Balanced Image", balanced_img)

# 等待用户按键
cv2.waitKey(0)
cv2.destroyAllWindows()

在这个例子中，我们创建了一个3x3的矩阵`gain`，其中第一行对应红色，第二行对应绿色，第三行对应蓝色。每个元素表示相应颜色通道的增益。然后，我们用`convertScaleAbs`函数将原始图像转换为增强后的图像。

请注意，这个方法的效果取决于你选择的增益值，实际应用中可能需要根据你的具体需求调整这些数值。如果你发现整个图像都偏绿，那么在`gain`矩阵中，绿色通道的增益值可能较高。