如何在沒有循環的python中將3d矩陣轉換爲3d乘法？

Question

我在做我的項目中的分割部分，我需要將3維蒙版乘以彩色圖像。

什麼現在做：

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 

是否有可能在沒有任何蟒循環繁殖？

Answer 1

AFAIK，numpy可以沿着顏色尺寸braodcastbiscuitMask以適應框架的形狀。
這個廣播是自動完成的。因此，

maskedFrame = rgbFrame * biscuitMask 
maskedHsvFrame = hsvFrame * biscuitMask 

應該沒有任何循環或修改。

PS，
如果你熟悉的matlab，廣播是非常喜歡運用bsxfun僅numpy的自動爲您完成。