matlab - Matlab：加快大型for循环

Question

我试图从一个大矩阵中得到 3 x 3 个掩码到向量中。目前，这是我的代码：

A=rand(3264,2448)

[rows cols]=size(A);

T=zeros(9,prod(size(A)-2));

for i=1:rows-2

  for j=1:cols-2

    T(:,(i-1)*cols+j)=reshape(A(i:i+2,j:j+2),[9 1]);

  end

end

目前，这需要相当长的时间。我可以以任何方式加快它吗？我在考虑meshgrid 和arrayfun，但我无法完全理解它们。

谢谢！

Answer 1

您可以使用im2col将图像块重新排列成列，例如：

T =  im2col(A,[3 3],'sliding');