白平衡和颜色转换

Question

我是一名编码初学者，我找到了用于调整图像白平衡的代码。代码有效，颜色正在改变。 实际上，我拍摄的所有图像的白平衡都有问题，我想用一个简单的代码来调整它。我使用的代码似乎效果最好（我尝试了在互联网上找到的不同代码）。 问题是河流区域在改造后变成了红色。我不知道这是因为白平衡还是因为我的颜色转换错误。 谁能帮帮我。

这是我使用的代码

import cv2 as cv
# import numpy as np

#
img = cv.imread("C:\\Users\\die5k\\Desktop\\NeuerOrdner\\DJI_0207.jpg")
r, g, b = cv.split(img)
r_avg = cv.mean(r)[0]
g_avg = cv.mean(g)[0]
b_avg = cv.mean(b)[0]

 # Find the gain of each channel
k = (r_avg + g_avg + b_avg) / 3
kr = k / r_avg
kg = k / g_avg
kb = k / b_avg

r = cv.addWeighted(src1=r, alpha=kr, src2=0, beta=0, gamma=0)
g = cv.addWeighted(src1=g, alpha=kg, src2=0, beta=0, gamma=0)
b = cv.addWeighted(src1=b, alpha=kb, src2=0, beta=0, gamma=0)

balance_img = cv.merge([b, g, r])
balance_img = cv.cvtColor(balance_img, cv.COLOR_BGR2RGB)
cv.imwrite('result0207.jpg', balance_img)

这是我的输入图像 这是我的结果

Answer 1

OpenCV 使用 BGR 排序。改变这个：

r, g, b = cv.split(img)

到这里：

b, g, r = cv.split(img)

然后删除这个：

balance_img = cv.cvtColor(balance_img, cv.COLOR_BGR2RGB)

因为OpenCV在保存文件时知道要转换为RGB。