首页 > 解决方案 > 在 Keras 中是否有可能具有宽度和高度为 32x32 的 input_shape?

问题描述

我正在使用带有 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)) 

任何想法都值得赞赏。

标签: pythontensorflowkerasvgg-net

解决方案


推荐阅读