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

if K.image_dim_ordering() == 'tf':
AttributeError: module 'keras.backend' has no attribute 'image_dim_ordering'

keras版本的问题,可以将语句修改为:

if K.image_data_format() == 'channels_first':

该参数是Keras 1.x中的image_dim_ordering,“channels_last”对应原本的“tf”,“channels_first”对应原本的“th”。以128x128x128的数据为例,“channels_first”应将数据组织为(3,128,128,128),而“channels_last”应将数据组织为(128,128,128,3)。该参数的默认值是~/.keras/keras.json中设置的值,若从未设置过,则为“channels_last”。
 

你可能感兴趣的:(tensorflow)