如何在不同大小的numpy数组之间应用逻辑运算?

mftmpeh8  于 2023-08-05  发布在  其他
关注(0)|答案(1)|浏览(77)

我有一个图像存储为一个numpy数组,形状为(x,y,3)。我有一个掩码,它是形状为(x,y)的numpy ndarray。我如何将这个蒙版应用到我的图像上?cv2.bitwise_and()似乎不起作用,直接用image[mask]索引也不起作用。具体地说,我的蒙版包含True和False,我想执行image & mask,但一次应用到所有层,而不是一次执行一层(R,G,B),然后再组合它们。

oxiaedzo

oxiaedzo1#

尝试使用mask[:.:,None]。这使得mask成为一个维数为(x,y,1)的数组,而普通的广播规则可以处理(x,y,1)与(x,y,3)的AND运算。

相关问题