我正在研究微调示例(目前在VGG16上)。我想用VGG16训练CIFAR10数据集,但预期的输入尺寸大于48x48px(CIFAR10有32x32)。
我找不到一种方法来调整图像大小以适应网络。
请帮帮我!
答案 0 :(得分:1)
您只需将input_shape
设置为您选择的尺寸即可。
请注意,由于VGG16预计至少48x48px,因此很可能会得到较差的结果。引自Keras documentation:
“...宽度和高度不应小于48.例如,(200,200,3)将是一个有效值。”
另一种方法是使用flow_from_directory
,您可以选择将图片大小调整为您想要的任何尺寸: