报错原因:高版本的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’
keras.backend.image_dim_ordering()
最新推荐文章于 2021-12-17 15:27:15 发布