matlab代码 MSRCR算法

以下是一个简单的 MATLAB 实现 MSRCR 算法的代码：

function output = MSRCR(input, scales, weight, radius)

% input: 输入图像
% scales: 尺度个数
% weight: 加权方式
% radius: 高斯半径

if nargin < 4
    radius = 3;
end
if nargin < 3
    weight = 'uniform';
end

% 将输入图像转换为 YCbCr 颜色空间
YCbCr = rgb2ycbcr(input);
Y = YCbCr(:,:,1);

% 计算输入图像的尺寸
[height, width, ~] = size(input);

% 计算每个尺度的标准差
stds = zeros(1, scales);
for i = 1:scales
    stds(i) = sqrt(2^(i-1)-1);
end

% 初始化输出图像
output = zeros(height, width);

% 对每个尺度进行处理
for i = 1:scales
    % 计算高斯核
    h = fspecial('gaussian', [radius*2+1, radius*2+1], stds(i));
    
    % 对输入图像进行模糊
    blurred = imfilter(Y, h, 'replicate');
    
    % 进行细节增强
    detail = Y - blurred;
    
    % 对细节图像进行放大和缩小
    detail_up = imresize(detail, 2, 'bilinear');
    detail_down = imresize(detail, 0.5, 'bilinear');
    
    % 对缩小的细节图像进行模糊
    blurred_down = imfilter(detail_down, h, 'replicate');
    
    % 计算权重
    if strcmp(weight, 'uniform')
        w = ones(size(detail));
    elseif strcmp(weight, 'gaussian')
        w = imfilter(abs(detail), h, 'replicate');
        w = 1 ./ (w.^2 + eps);
    end
    
    % 进行细节融合
    detail = (detail_up - imresize(blurred_down, [height, width], 'bilinear')) .* w;
    
    % 更新输出图像
    output = output + detail;
end

% 将输出图像转换回 RGB 颜色空间
output = YCbCr;
output(:,:,1) = output;

output = ycbcr2rgb(output);

% 将输出图像限制在 0 到 1 之间
output(output < 0) = 0;
output(output > 1) = 1;

end

这段代码实现了 MSRCR 算法，包括对每个尺度的处理、权重计算和细节融合等步骤。函数接受输入图像、尺度个数、加权方式和高斯半径等参数，并返回增强后的输出图像。在代码中，我们使用了 MATLAB 自带的函数 `fspecial` 和 `imfilter` 来实现高斯核的生成和图像的滤波。同时，我们还使用了 `imresize` 函数来实现图像的放大和缩小。