我正在使用带有 Keras 和 Tensorflow 的 Python 作为后端,我想为我的模型使用尽可能小的输入图像。
VGG19 应用程序表示它允许输入形状的宽度和高度低至 32。几周前,Keras 网站上的最小值为 48。我用 48 没有问题。现在,我想尝试 32,但我不能这样做,因为我得到:
ValueError: Input size must be at least 48x48; got `input_shape=(32, 32, 3)`
我提到我将 Keras 更新到了 2.2.2 版本,并且我还删除了 VGG19 的权重模型文件(来自~/.keras/models/
)。如果我将 input_shape 设置为 (48, 48, 3),它会再次下载模型,否则会出现上述错误。
简而言之,我的代码只想加载预训练模型:
vgg_conv = VGG19(weights='imagenet', include_top=False, input_shape=(32, 32, 3))
任何想法都值得赞赏。