请写出计算SSIM的matlab完整代码

Sure! 以下是计算SSIM的MATLAB代码：

function [ssimval, ssimmap] = ssim(A, B)

% 确保输入的图像是灰度图像
if size(A,3) > 1
    A = rgb2gray(A);
end

if size(B,3) > 1
    B = rgb2gray(B);
end

% 常数定义
K1 = 0.01;
K2 = 0.03;
L = 255;
C1 = (K1*L)^2;
C2 = (K2*L)^2;

% 均值和标准差计算
mu_A = mean2(A);
mu_B = mean2(B);
sigma_A = std2(A);
sigma_B = std2(B);
sigma_AB = cov(A,B);

% 计算 SSIM
numerator = (2*mu_A*mu_B + C1)*(2*sigma_AB + C2);
denominator = (mu_A^2 + mu_B^2 + C1)*(sigma_A^2 + sigma_B^2 + C2);

ssimval = numerator/denominator;
ssimmap = ssimval;

end

这个函数接受两个输入图像 A 和 B，并返回一个 SSIM 值（在 0 到 1 之间）和一个 SSIM 地图（用于显示哪些区域的图像质量更高）。需要注意的是，该函数假定输入的图像是灰度图像。如果输入的图像是彩色图像，则需要使用 `rgb2gray` 函数将其转换为灰度图像。