4

我有一个大矩阵，我想从中收集一组子矩阵。如果我的矩阵是 NxN 并且子矩阵大小是 MxM，我想收集I=(N - M + 1)^2子矩阵。换句话说，我希望原始矩阵中的每个元素都有一个 MxM 子矩阵，这些元素可以位于此类矩阵的左上角。

这是我的代码：

for y = 1:I
    for x = 1:I
        index = (y - 1) * I + x;
        block_set(index) = big_mat(x:x+M-1, y:y+M-1)
    endfor
 endfor

输出如果 a) 错误，并且 b) 暗示big_mat(x:x+M-1, y:y+M-1)表达式中有一些东西可以让我得到我想要的东西，而不需要两个 for 循环。任何帮助将非常感激

4

3 回答 3

5

您的代码中似乎有一些问题。如果我要使用双循环，我会这样做：

M = someNumber;
N = size(big_mat,1); %# I assume big_mat is square here

%# you need different variables for maxCornerCoord and nSubMatrices (your I)
%# otherwise, you are going to index outside the image in the loops!
maxCornerCoord = N-M+1;
nSubMatrices = maxCornerCoord^2;

%# if you want a vector of submatrices, you have to use a cell array...
block_set = cell(nSubMatrices,1); 
%# ...or a M-by-M-by-nSubMatrices array...
block_set = zeros(M,M,nSubMatrices);
%# ...or a nSubMatrices-by-M^2 array
block_set = zeros(nSubMatrices,M^2);

for y = 1:maxCornerCoord
    for x = 1:maxCornerCoord
        index = (y - 1) * maxCornerCoord + x; 
        %# use this line if block_set is a cell array
        block_set{index} = big_mat(x:x+M-1, y:y+M-1);
        %# use this line if block_set is a M-by-M-by-nSubMatrices array
        block_set(:,:,index) = big_mat(x:x+M-1, y:y+M-1);
        %# use this line if block_set is a nSubMatrices-by-M^2 array
        block_set(index,:) = reshape(big_mat(x:x+M-1, y:y+M-1),1,M^2);
    endfor
 endfor

编辑

我刚刚看到有一个用于 Octave的im2col实现。因此，您可以将双循环重写为

%# block_set is a M^2-by-nSubMatrices array
block_set = im2col(big_mat,[M,M],'sliding');

%# if you want, you can reshape the result to a M-by-M-by-nSubMatrices array
block_set = reshape(block_set,M,M,[]);

这可能更快，并且节省了大量的数字树。

于 2010-04-26T21:33:48.253 回答
1

使用 Mathematica：此代码创建一个矩阵，其中每个元素都是 MxM 的矩阵，原始矩阵中的每个元素都位于此类矩阵的左上角。

右侧和底部的矩阵元素用 x 填充。

Partition[big_mat, {M, M}, {1, 1}, {1, 1}, x]

示例： 替代文字 http://img130.imageshack.us/img130/6203/partitionf.png

如果您不使用 x 参数，那么它会定期自动采样。

于 2010-04-26T23:51:04.757 回答
0

关于您的输出错误，可能是因为分配。您正在尝试将矩阵分配给向量位置。尝试使用

block_set(:,:,index) = big_mat(x:x+M-1, y:y+M-1)

反而。

于 2010-04-26T20:39:43.460 回答