如何在matlab中使用蒙版平滑3D矩阵

Question

我需要平滑一个 3D 矩阵 M。平滑的输出为S。 matlab代码可以是这样的：

S = smooth3(M, 'box', 3); 

问题只是矩阵M中的某些部分在平滑过程中应该被考虑。我有一个面具A 来识别这些部件的位置。 A与M的大小相同，1的值表示考虑这一点，0表示忽略这一点。该怎么做？

例如，如果我在下面有M 和A：

M=zeros(7,7,7); M(2:6,2:6,2:6)=ones(5,5,5); M(4,4,4)=28;

A=zeros(7,7,7); A(2:6,2:6,2:6)=ones(5,5,5);

平滑之后，我想要这样的结果：

S=zeros(7,7,7); S(2:6,2:6,2:6)=smooth3(M(2:6,2:6,2:6), 'box', 3); 

任何人都知道如何在 matlab 中使用掩码进行这种平滑？提前谢谢你。

Answer 1

我找到了解决办法：

index=find(A==0);
M(index) = 0;

k=ones(3,3,3);
counts = convn(A,k,'same');
sums = convn(M,k,'same');

S= sums ./counts .* A;

我测试过了。好像没问题。