matlab中total_variation函数的代码

以下是matlab中实现全变分（Total Variation）的total_variation函数的代码：

function [output_image] = total_variation(input_image, regularization_type, regularization_strength)
% input_image: 输入图像
% regularization_type: 正则化项类型，可选'L1'或'L2'
% regularization_strength: 正则化强度

% 将输入图像转换为double类型
input_image = im2double(input_image);

% 定义正则化矩阵
switch regularization_type
    case 'L1'
        regularization_matrix = [0 -1 0; -1 4 -1; 0 -1 0];
    case 'L2'
        regularization_matrix = [0 1 0; 1 -4 1; 0 1 0];
    otherwise
        error('Invalid regularization type.');
end

% 计算梯度
gx = conv2(input_image, [-1 1], 'same');
gy = conv2(input_image, [-1; 1], 'same');

% 计算正则化项
if regularization_strength > 0
    reg_term = regularization_strength * conv2(input_image, regularization_matrix, 'same');
else
    reg_term = 0;
end

% 计算全变分图像
output_image = input_image + gx + gy + reg_term;
end

该函数首先将输入图像转换为double类型，然后根据正则化项类型计算正则化矩阵。接着，使用卷积运算计算图像的梯度，并根据正则化强度计算正则化项。最后，将梯度、正则化项和输入图像相加得到全变分图像。