matlab中18张jpg彩色图片的全变分图像融合算法代码

全变分图像融合算法可以实现多张彩色图片的自然融合，保持每张图片的特点，同时融合出一张自然细腻的彩色图像。下面是一份基于matlab的18张jpg彩色图片的全变分图像融合算法的实现代码。

%% 读入18张彩色图片
for i = 1:18
    img = imread(['img', num2str(i), '.jpg']);
    img = imresize(img, [500 500]); % 调整图片大小以便后续处理
    if size(img, 3) == 3
        gray_img(:,:,i) = rgb2gray(img);
    else
        gray_img(:,:,i) = img;
    end
    color_img(:,:,:,i) = img;
end

%% 用全变分方法实现图像融合
lambda = 5e-3; % 正则化参数
tol = 1e-3; % 迭代停止误差
maxIter = 100; % 最大迭代次数
[m, n] = size(gray_img(:,:,1));
numImgs = size(gray_img, 3);

% 构造矩阵Dx和Dy
Dx = spdiags([repmat([-1, 1], 1, n-1); zeros(1, n)], [0,1], n, n);
Dy = spdiags([repmat([-1;1], m-1, 1), zeros(m, 1)], [0, m], m*n, m*n);

% 计算全变分图像融合结果
fused_img = zeros(m, n, 3);
for k = 1:3 % 分别对RGB三个通道进行处理
    color_img_k = color_img(:,:,k,:);
    color_img_k = reshape(color_img_k, [], numImgs);
    fused_img_k = zeros(m*n, 1);
    for i = 1:maxIter
        fused_img_k_old = fused_img_k;
        fused_img_k = zeros(m*n, 1);
        for j = 1:numImgs
            gray_img_j = gray_img(:,:,j);
            color_img_k_j = color_img_k(:,j);
            A = [speye(m*n) + lambda * (Dx'*Dx + Dy'*Dy); sqrt(lambda) * gray_img_j(:)];
            b = [sqrt(lambda) * color_img_k_j(:); zeros(m*n, 1)];
            fused_img_k = fused_img_k + lsqr(A, b, tol, 5000);
        end
        if norm(fused_img_k - fused_img_k_old) / norm(fused_img_k_old) < tol
            break;
        end
    end
    fused_img(:,:,k) = reshape(fused_img_k, m, n);
end

% 显示全变分图像融合结果
imshow(uint8(fused_img));

需要注意的是，全变分图像融合算法的计算量较大，同时需要较长的计算时间，因此建议在较高配置的计算机上进行实现。