Optimizing: Getting rid of for loop
显示 更早的评论
Hi, I've been scratching my head on this problem and keep seem to figureout a good way to do it.
I have a big matrix : size(A)=1024x1024x400 where the 3rd dimension is that of time and the two first of space. Here is an example of what I would want to do with it:
for x=1:1022
for y=1:1022
B(y,x,1) = mean2(A(y:y+2,x:x+2,1));
end
end
Doing it like that works but Matlab will do the calculations on one element at the time instead of all at the same time like in the following case:
B(:,:,1) = A(:,:,1).*A(:,:,2);
Is there a way to rewrite my initial code or use a function that would make it way faster? Thank you for any help.
采纳的回答
Not sure what the mean2d function do as it's not a standard matlab function. Assuming it's calculating a mean, then probably
kernel = ones(3, 3, size(A, 3));
kernel = kernel ./ numel(kernel); %convolution kernel for mean
B = convn(A, kernel, 'valid');
2 个评论
Raphaël
2017-12-5
Because of the separable nature of the kernel, it may be more efficient to do this in separable passes,
B=convn(A,[1,1,1]/9, 'valid');
B=convn(B,[1;1;1],'valid');
更多回答（0 个）
类别
在 帮助中心 和 File Exchange 中查找有关 Problem-Based Optimization Setup 的更多信息
Translated by 
