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/