如何在python中实现无循环的3d矩阵到3d乘法？

Question

我在我的项目中做分割部分，我需要将一个三维掩模乘以一个彩色图像。在

现在在做什么：

maskedFrame=np.zeros((rgbFrame.shape))
maskedHsvFrame=np.zeros((rgbFrame.shape))
for color in range(0,3):
    maskedFrame[:,:,color]=rgbFrame[:,:,color]*biscuitMask
    maskedHsvFrame[:,:,color]=hsvFrame[:,:,color]*biscuitMask

在python中不使用任何循环就可以进行乘法吗？在

Answer 1

另外，numpy可以沿着颜色维度braodcastbiscuitMask来适应框架的形状。
这个广播是自动进行的。因此

maskedFrame = rgbFrame * biscuitMask
maskedHsvFrame = hsvFrame * biscuitMask 

应该在没有任何循环或修改的情况下工作。在

PS，
如果您熟悉matlab，广播非常像应用bsxfun，只有numpy会自动为您完成。在