Matlab 3D矩阵通过2D矩阵矢量化对三维进行转换

Question

我有3D矩阵A的大小N, M, 3和2D变换T大小的3, 3 ，我适用于矩阵的第三维A 。 如果相关，此3D矩阵是RGB图像。

我当前的解决方案如下所示：

for i = 1:N
    for j = 1:M
        A(i, j, :) = T * [A(i, j, r); A(i, j, g); A(i, j, b)];
    end
end

但是我很好奇是否有矢量化的方法吗？

Answer 1

你可以试试看 首先切换尺寸，使通道成为第一个：

B = permute(A,[3,1,2])

然后将B整形为大小为3xN * M的矩阵：

C = reshape(B,3,[])

现在计算一个矩阵乘法：

D = T*C;

然后，您需要重塑和重新排列：

E = permute(reshape(D,3,M,N),[2,3,1])

我没有对其进行测试，因此可能缺少一些东西，但是通常这应该可以工作（而且速度很快！）