Mar*_*k.F 6 mask computer-vision conv-neural-network keras
我有一个RGB形状的图像，(256,256,3)并且我有一个形状的权重蒙版(256,256)。如何使用Keras在它们之间进行逐元素乘法？（所有通道共享相同的掩码）
您需要一个，Reshape以便两个张量都具有相同数量的维，并且一个Multiply层
mask = Reshape((256,256,1))(mask)
out = Multiply()([image,mask])
如果形状可变，则可以使用单个Lambda图层，如下所示：
mask = Reshape((256,256,1))(mask)
out = Multiply()([image,mask])
| 归档时间： |
|
| 查看次数： |
5166 次 |
| 最近记录： |