我的模型中内置了几个图像预处理步骤:
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中有一个很好的方法吗?
答案 0 :(得分:1)
您可以使用:model.layers[index].output
轻松获取任何图层的输出,并使用K.function
从中创建张量函数,然后评估函数并可视化输出。