引用github中提到的答案,您需要指定维度顺序:
Keras是Theano或Tensorflow库的包装器。 image_dim_ordering
使用设置变量image_dim_ordering
来确定输入图层是Theano还是Tensorflow格式。可以通过两种方式指定此设置-
- 在
~/.keras/keras.json
指定'tf'
或'th'
,例如image_dim_ordering: 'th'
。注意:这是一个json文件 。 - 或像这样在模型中指定
image_dim_ordering
:model.add(MaxPooling2D(pool_size=(2, 2), dim_ordering="th"))
附录: image_dim_ordering
在'th'
模式下,通道尺寸(深度)在索引1(例如image_dim_ordering
)处。在'tf'
模式下,它位于索引3(例如256、256、3)。在评论中引用@naoko。
0
我正在尝试在keras中复制VGG16模型,以下是我的代码:
maxpooling2d层在注释的行上给出错误
错误提示:
这可能是什么原因?如何解决呢?
编辑:更详细的错误日志: