使用 Tensorflow 的 Keras 代码存在问题。
我有一个简单的网络,我需要在输入后立即进行元素乘法。这部分代码如下所示。我总是收到错误:
*** AttributeError: 'NoneType' 对象没有属性 '_inbound_nodes'
我的代码:
input_img = Input(shape=(256, 256, 2, 1, 2))
masked = Lambda(lambda x: tf.multiply(x[0], x[1]))([input_img[:,:,:,:,:,0], input_img[:,:,:,:,:,1]])
conv1 = Conv3D(1, (5, 5, 2), padding='same', activation=activation, kernel_regularizer=regularizers.l2(1e-9))(masked)
net_head = Model(inputs=input_img, outputs=conv1)
我已经使用了 Lambda 层,所以我很困惑是什么导致了这个问题。输入的维度很适合模型,所以这应该不是问题。
有人有线索吗?
这种逐元素乘法实际上是将一些像素屏蔽为零。在上面的代码中,掩码在附加的最后一个维度中连接到输入变量中。或者,我还尝试使用掩码专用的变量,例如 input_mask,而不是将其作为 input_image 中的附加最后一个维度包含在内,以进行乘法运算,这对相同的错误信息也不起作用。
我必须在网络的输入层之后而不是在输入层之前对输入数据进行掩码处理的原因是,我需要输入图像的完整数据(包括被掩码)在使用shuffle进行损失计算在。
DIEA
蝴蝶不菲
相关分类