color balance

%白平衡
clc
close all
clear 
 RGB=imread('BW1.png');
%  RGB=imread('BW2.png');
%  RGB=imread('BW3.jpg');
figure,imshow(RGB);
R = RGB(:,:,1);      G = RGB(:,:,2);      B = RGB(:,:,3);
I=0.299*R+0.587*G+0.114*B;
I_mean=mean(I(:));
R_mean=mean(R(:));
G_mean=mean(G(:));
B_mean=mean(B(:));

kR=I_mean/R_mean;
kG=I_mean/G_mean;
kB=I_mean/B_mean;

gR=kR*R
gG=kG*G;
gB=kB*B;

f=cat(3,gR,gG,gB);
figure,imshow(uint8(f));

%最大值平衡
Rmax=max(R(:));
Gmax=max(G(:));
Bmax=max(B(:));
SRGB=min(min(Rmax,Gmax),Bmax);

nr=find(R(:)>=SRGB);
Nr=length(nr);

ng=find(G(:)>=SRGB);
Ng=length(ng);

nb=find(B(:)>=SRGB);
Nb=length(nb);

Nmax=max(max(Nr,Ng),Nb);
tr=sort(R(:),'descend');
Tr=tr(Nmax);
tg=sort(G(:),'descend');
Tg=tg(Nmax);
tb=sort(B(:),'descend');
Tb=tb(Nmax);

kr=SRGB/Tr;
kg=SRGB/Tg;
kb=SRGB/Tb;

gR2=kr*R;
gG2=kg*G;
gB2=kb*B;


f2=cat(3,gR2,gG2,gB2);
figure,imshow(uint8(f2));