问题
原始代码如下:
keras.backend.set_image_dim_ordering('th')
运行出现报错:
AttributeError: module 'keras.backend' has no attribute 'set_image_dim_ordering'
原因
在Keras最新版本中已经把 image_dim_ordering 改为 image_data_format
解决
修改后的代码为:
keras.backend.set_image_data_format('channels_first')
报错消失!
参考:
1.Error : keras.backend has no attribute ‘set_image_dim_ordering’