帮我写matlab计算图像mtf值的函数

当我们谈论图像的MTF（Modulation Transfer Function）时，我们实际上是在谈论图像系统对各种空间频率的信号的传输能力。MTF是用于描述图像系统传输能力的一个常用参数。

下面是一个使用Matlab计算图像MTF值的函数。

function mtf = calculateMTF(image)
% image: 输入的图像

% 定义一些参数
numStages = 10; % 傅里叶变换的级数
samplingFrequency = 2; % 采样频率
minFrequency = 0.1; % 最小频率
maxFrequency = 1.0; % 最大频率

% 转换为灰度图像
if size(image, 3) == 3
    image = rgb2gray(image);
end

% 计算图像的傅里叶变换
ft = fftshift(fft2(double(image)));

% 将傅里叶变换转换为幅度谱
amplitude = abs(ft);

% 计算频率空间
[rows, cols] = size(image);
x = linspace(-cols/2, cols/2-1, cols) / (cols/samplingFrequency);
y = linspace(-rows/2, rows/2-1, rows) / (rows/samplingFrequency);
[X, Y] = meshgrid(x, y);
distance = sqrt(X.^2 + Y.^2);

% 计算MTF
mtf = zeros(numStages, 1);
for stage = 1:numStages
    % 计算该级的频率
    frequency = (maxFrequency - minFrequency) / numStages * stage + minFrequency;
    
    % 计算该级的带宽
    bandwidth = frequency / 2; 
    
    % 计算该级的低通滤波器
    lowpass = exp(-(distance / bandwidth).^2);
    
    % 计算该级的高通滤波器
    highpass = 1 - lowpass;
    
    % 计算该级的MTF
    mtf(stage) = mean2(highpass .* amplitude) / mean2(amplitude);
end

% 可视化MTF曲线
plot(mtf, 'LineWidth', 2);
xlabel('MTF Stage');
ylabel('MTF Value');
title('Modulation Transfer Function (MTF)');

这个函数的输入是一个图像，输出是图像的MTF曲线。函数首先将图像转换为灰度图像，然后计算图像的傅里叶变换。接下来，函数计算频率空间，并使用一系列低通滤波器和高通滤波器来计算图像的MTF。最后，函数可视化MTF曲线。