白平衡

最新推荐文章于 2023-03-22 23:43:09 发布

原创最新推荐文章于 2023-03-22 23:43:09 发布 · 321 阅读

0 ·

CC 4.0 BY-SA版权

python 同时被 2 个专栏收录

20 篇文章

订阅专栏

opencv

9 篇文章

订阅专栏

本文介绍了一种基于平均亮度的图像白平衡校正方法。该方法通过计算图像的RGB通道平均值，并调整各通道的增益系数来实现颜色校正，使图像看起来更加自然。代码实现了图像读取、尺寸调整、白平衡校正及显示等功能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

# -*- coding: utf-8 -*-
"""
Created on Fri Jun 22 19:00:54 2018

@author: PC
"""
import cv2
import numpy as np

def whiteBalance(src):
    B,G,R = cv2.split(src)
    B_mean = np.mean(B)
    G_mean = np.mean(G)
    R_mean = np.mean(R)
    gray_mean = (B_mean+G_mean+R_mean) / 3
    B_corf = 0.9  * gray_mean / B_mean
    G_corf = 0.9 * gray_mean / G_mean
    R_corf = 0.9 * gray_mean / R_mean
    
    B_Correct = np.uint8(B_corf * B)
    G_Correct = np.uint8(G_corf * G)
    R_Correct = np.uint8(R_corf * R)
    
    dst = cv2.merge((B_Correct,G_Correct,R_Correct))
    
    return dst

def main():
    img = cv2.imread('C:\\Users\\PC\\Desktop\\program\\skin\\Images\\12345\\4.jpg')
    width, height = img.shape[:2]
    if height>800 :
        rate = height/800
        width1 = int(width/rate)
        height1 = 800
    img = cv2.resize(img,(height1,width1),cv2.INTER_CUBIC)
#    dst = LightNormalization.lightNormalization(img)
    dst = whiteBalance(img)
    cv2.imshow("img", img)
    cv2.imshow("whiteBalance", dst)
    cv2.waitKey(0)
    cv2.destroyAllWindows()
    
if __name__ == "__main__":
    main()