用matlab给我写一个网格法白平衡的程序

以下是一个简单的 MATLAB 程序，用于通过网格法实现白平衡：

% 读取图像
img = imread('input.jpg');

% 将图像转换为 double 类型，方便后续计算
img = im2double(img);

% 计算图像的 RGB 三通道均值
r_mean = mean(mean(img(:,:,1)));
g_mean = mean(mean(img(:,:,2)));
b_mean = mean(mean(img(:,:,3)));

% 计算白平衡矩阵
wb_matrix = [1/r_mean, 0, 0; 0, 1/g_mean, 0; 0, 0, 1/b_mean];

% 对图像进行白平衡校正
wb_img = zeros(size(img));
for i = 1:size(img,1)
    for j = 1:size(img,2)
        pixel = [img(i,j,1), img(i,j,2), img(i,j,3)];
        wb_pixel = wb_matrix * pixel';
        wb_img(i,j,:) = wb_pixel';
    end
end

% 显示原始图像和白平衡校正后的图像
figure;
subplot(1,2,1);
imshow(img);
title('原始图像');
subplot(1,2,2);
imshow(wb_img);
title('白平衡校正后的图像');

请注意，此程序仅仅是一个简单的示例，实际应用中可能需要更加复杂的算法和处理步骤。