图像白均衡

import cv2 as cv
if __name__ == '__main__':
    img = cv.imread('shu.png')
    r,g,b = cv.split(img)
    r_avg = cv.mean(r)[0]
    b_avg = cv.mean(b)[0]
    g_avg = cv.mean(g)[0]

    k = (r_avg+b_avg+g_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)

    img = cv.merge([r,g,b])
    cv.imshow("a",img)
    cv.imwrite("a.jpg",img)
    cv.waitKey(0)