限制 keras 层中的输出值答案

【问题标题】：restricting output values in keras layer限制 keras 层中的输出值
【发布时间】：2020-09-09 11:14:30
【问题描述】：

我已经写了这个NN

decoder_output = Conv2D(64, (3,3), activation='relu', padding='same')(encoder_input)
decoder_output = UpSampling2D((2, 2))(decoder_output)
decoder_output = Conv2D(32, (3,3), activation='relu', padding='same')(decoder_output)
decoder_output = UpSampling2D((2, 2))(decoder_output)
decoder_output = Conv2D(16, (3,3), activation='relu', padding='same')(decoder_output)
decoder_output = UpSampling2D((2, 2))(decoder_output)
decoder_output = Conv2D(2, (3, 3), activation='sigmoid', padding='same')(decoder_output)
decoder_output = UpSampling2D((2, 2))(decoder_output)
decoder_output = Flatten()(decoder_output)
decoder_output = Dense(height*width, activation='relu')(decoder_output)
model = Model(inputs=encoder_input, outputs=decoder_output)
model.compile(optimizer='adam', loss='mse')
clean_images = model.fit(train_images, y_train_red, epochs=10,validation_data=(validation_images,y_validation_red))

假设返回图像值。有没有办法将返回值限制为 int 和/或将输出层值最大化为 255？

【问题讨论】：

你可以关注this

标签： python tensorflow keras restriction

【解决方案1】：

应该发生的是，您的模型将学会不输出高于 255 和低于 0 的值。但是，在它这样做的情况下，您可以在预测时将值裁剪为 0 到 255 之间。关于整数输出，我不知道有一种方法。但是，您可以在预测时对输出进行四舍五入。

【讨论】：