模型获得正确的输入尺寸,但引发尺寸错误

时间:2019-05-24 11:57:37

标签: python tensorflow keras keras-2 tensorflow2.0

编辑:已解决。拼写错误,请参阅第一个评论


  • 我在Keras2 / TF2中使用MobileNetV2预训练模型
    • 我的图片缩放为224x224
    • 我设置了模型输入形状:input_shape=(224,224,3)

稍后,我使用ImageGenerator创建图像。在这里设置

  • data_format='channels_last'

因为模型需要在位置2(0,1,2)的通道。

然后在flow_from directory中设置

  • target_size=[224,224]

当我拟合模型时,出现以下错误:

  

ValueError:检查输入时出错:预期input_8具有形状   (224,224,3),但数组的形状为(244,244,3)


是否可以通过在Generator选项中使用channels_last来使flow_from_directory中的目标错误地将尺寸标识为通道?

或者检查错误回溯extract_tensors_from_dataset=True会从文件而不是我给定的input_shape中读取尺寸。在(keras1)手册中,其内容为:

  

...如果您要推断input_shape,也可以省略此选项   来自input_tensor。如果您选择同时包含input_tensor和   如果形状匹配,则使用input_shape,然后将使用input_shape   不匹配,那么我们将抛出错误。例如。 (160,160,3)将是   一个有效值...

对我来说,它看起来像是从张量中提取input_shape,然后由于形状不匹配而引发了错误……实际上是这样。

input_tensor也默认为None,但我没有更改。

有人知道这是怎么回事吗?

谢谢

0 个答案:

没有答案