keras.backend.image_dim_ordering()

报错原因:高版本的keras没有keras.backend.image_dim_ordering()
高版本用的是keras.backend.image_data_format()来判断图像数据格式,主要是判断通道在前还是在后。
tensorflow的通道在后,比如1000张1024*720的RGB图像,在tensorflow中为(1000,1024,720,3)
keras.backend.image_data_format() 返回默认图像数据格式约定 (‘channels_first’ 或 ‘channels_last’)。
解决方法:
将if K.image_dim_ordering() == ‘tf’:
改为if K.image_data_format() =='channels_last’

你可能感兴趣的:(tensorflow)