Keras CIFAR10微调VGG16:如何预处理输入数据以适应VGG16网络?

时间:2017-06-22 13:05:11

标签: python deep-learning keras

我正在研究微调示例(目前在VGG16上)。我想用VGG16训练CIFAR10数据集,但预期的输入尺寸大于48x48px(CIFAR10有32x32)。

我找不到一种方法来调整图像大小以适应网络。

请帮帮我!

1 个答案:

答案 0 :(得分:1)

您只需将input_shape设置为您选择的尺寸即可。

请注意,由于VGG16预计至少48x48px,因此很可能会得到较差的结果。引自Keras documentation

“...宽度和高度不应小于48.例如,(200,200,3)将是一个有效值。”

另一种方法是使用flow_from_directory,您可以选择将图片大小调整为您想要的任何尺寸:

https://keras.io/preprocessing/image/