image_dim_ordering()

函数描述:
返回默认的图像的维度顺序(‘tf’或‘th’)
对2D数据来说:
tf假定维度顺序为(rows,cols,channels)
th假定维度顺序为(channels, rows, cols)
对3D数据而言:
tf假定(conv_dim1, conv_dim2, conv_dim3, channels)
th则是(channels, conv_dim1, conv_dim2, conv_dim3)

正是因为keras封装了两种后端引擎Theano/Tensorflow,通过将二者的函数统一封装,使得用户可以以同一接口调用不同的后端引擎的函数。上述的问题就是tf维度是在最后的,而th是在最前面,所以每次会判断一下。

你可能感兴趣的:(Tensorflow学习)