可视化keras中预处理图层的结果

时间:2017-01-29 08:08:17

标签: keras keras-layer

我的模型中内置了几个图像预处理步骤:

model = Sequential()
model.add(Cropping2D(cropping=((22, 0), (0, 0)), input_shape=(160, 320, 3)))
model.add(Lambda(lambda image: tf.image.resize_images(image, (66, 200))))
model.add(Lambda(lambda image: image / 255.0 - 0.5))

这种方法的好处是培训和测试的管道完全相同。缺点是很难验证预处理步骤的结果,例如,通过观察裁剪步骤后的图像。

在keras中有一个很好的方法吗?

1 个答案:

答案 0 :(得分:1)

您可以使用:model.layers[index].output轻松获取任何图层的输出,并使用K.function从中创建张量函数,然后评估函数并可视化输出。

检查: Keras, How to get the output of each layer?