keras - 使用 Keras 进行元素乘法

Question

我有一个形状为 (256,256,3) 的 RGB 图像我有一个形状为 (256,256) 的重量面具.如何使用 Keras 执行它们之间的逐元素乘法？ (所有 channel 共享相同的掩码)

Answer 1

您需要一个 Reshape所以两个张量都有相同的维数，还有一个 Multiply层

mask = Reshape((256,256,1))(mask) 
out = Multiply()([image,mask])

如果您有可变形状，则可以使用单个 Lambda像这样的层:

import keras.backend as K 

def multiply(x):
    image,mask = x
    mask = K.expand_dims(mask, axis=-1) #could be K.stack([mask]*3, axis=-1) too 
    return mask*image

out = Lambda(multiply)([image,mask])