VGG16 Keras CIFAR10微调：我如何预处理输入数据以适应VGG16网络？

Question

我正在进行微调示例（目前在VGG16上）。我想用VGG16训练CIFAR10数据集，但预期的输入尺寸大于48x48像素（CIFAR10具有32x32）。

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

请帮我一把！

Answer 1

您可以简单地将input_shape设置为您选择的维度。

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

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

另一种方法是使用flow_from_directory，让您调整图片大小，以任何你想要的尺寸的选项：

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