如何在python中将二进制图像t rgb图像相乘？

Question

我有一个二进制图像，它是另一个彩色图像的分段形式。

如您所知，二进制图像是2-d，而RGB图像是3-d，如何将它们相乘？请希望

Answer 1

将图像和蒙版转换为numpy数组。 使用numpy数组进行元素明智的乘法无需任何特殊处理即可完成。例如：

a = np.random.randint(0,10,(3,2,2)) # RGB of size 2x2
b = np.random.randint(0,2,(2,2))    # Binary mask of size 2x2
c = a*b

输出：

a = array([ [[7, 6],
             [5, 8]],

            [[1, 3],
             [8, 5]],

            [[1, 8],
             [4, 4]]])

b = array(  [[1, 0],
             [0, 1]])

c = array([ [[7, 0],
             [0, 8]],

            [[1, 0],
             [0, 5]],

            [[1, 0],
             [0, 4]]])