Matlab:加速大型for循环

Question

我正在尝试将一个大矩阵中的3x3个掩码转换为向量。目前，这是我的代码：

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

目前，这需要相当长的时间。我能以任何方式加快速度吗？我在想网格和数组的乐趣，但是我不能完全理解它们。

谢谢!

Answer 1

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

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