AttributeError: 模块“keras.backend”没有属性“image_dim_ordering”

Question

我尝试执行一些教程迁移学习项目。但我有属性错误。

我检查了我的 tensorflow 和 keras 版本。

张量流：1.14.0 keras：2.2.5

和 python 3.6.9 版本。

代码在这里。

if(K.image_dim_ordering() == 'th'):
  input_tensor = Input(shape=(3, 299, 299))

错误信息在这里。

AttributeError: module 'keras.backend' has no attribute 'image_dim_ordering'

Answer 1

keras.backend.common 模块有 image_dim_ordering()

if(K.common.image_dim_ordering() == 'th'):

    input_tensor = Input(shape=(3, 299, 299))

Answer 2

替换image_dim_ordering为image_data_format

if(K.image_dim_ordering() == 'th'): 
  input_tensor = Input(shape=(3, 299, 299))

将上面的代码改为

if K.image_data_format() == 'th':
  input_tensor = Input(shape=(3, 299, 299))

Answer 3

与我有同样问题的人的自我回答。image_dim_ordering 从 Keras 2.x 更改为 image_data_format。由于将 image_dim_ordering 更改为 image_data_format。

参考链接