我正在尝试使用MATLAB通过两种方法对具有高斯滤波器的图像进行卷积:使用1D FFT的可分卷积和使用2D FFT的不可分卷积。我期望可分离的卷积更快。但是,它不是用于较小的图像,而是用于2D更快的较大图像。我不确定这是我的实现问题还是因为我的概念不正确。
这是我的代码:
img1 = randi([1,256],128,128);
% Create a Gaassian filter
rf1 = fspecial('gaussian', [1 128], 1.0);
cf1 = transpose(rf1);
gf1 = cf1 * rf1;
rc1 = round(conv2(img1, gf1, 'same'));
rc1 = fft2dconv(img1, gf1);
rc2 = fft1dconv(img1, rf1, cf1);
function o = fft1dconv(img, rowf, colf)
% Zero-Pad
imgsize = size(img);
rsize = size(rowf);
csize = size(colf);
img = padarray(img, [imgsize(1)/2, imgsize(2)/2]);
rowf = padarray(rowf, [2*imgsize(1)-rsize(1), 2*imgsize(2)-rsize(2)], 'post');
colf = padarray(colf, [2*imgsize(1)-csize(1), 2*imgsize(2)-csize(2)], 'post');
% Seperable convolution using 1D FFT
tic;
result = fft(transpose(fft(img))) .* fft(transpose(fft(colf)));
result = result .* fft(transpose(fft(rowf)));
o = transpose(round(real(ifft2(result))));
toc;
% Remove Pad
o = o(imgsize(1)+1:2*imgsize(1),imgsize(2)+1:2*imgsize(2));
end
function o = fft2dconv(img, filter)
%zero-pad
imgsize = size(img);
fsize = size(filter);
img = padarray(img, [imgsize(1)/2, imgsize(2)/2]);
filter = padarray(filter, [2*imgsize(1)-fsize(1), 2*imgsize(2)-fsize(2)], 'post');
% Non-Seperable convolution using 2D FFT
tic;
o = round(real(ifft2(fft2(img) .* fft2(filter))));
toc;
% Remove Pad
o = o(imgsize(1)+1:2*imgsize(1),imgsize(2)+1:2*imgsize(2));
end
计时结果为:
Elapsed time is 0.003315 seconds.
Elapsed time is 0.004369 seconds.
对于4 x 4的图像,可分离的方法要快得多,但是对于较大的图像。事实并非如此,我不确定为什么。任何帮助将非常感激。
我建议您分析您的代码以查看发生了什么。您可能正在执行其他一些操作,而不是进行核心FFT计算本身。
进入MATLAB并输入profile viewer
。完成后,在该窗口中运行命令并完成它。完成后,它将确定您代码的密集部分,您可以从那里找出如何对其进行优化。您说对了,可分离的过滤器更快。
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句