TensorFlow函数教程：tf.keras.backend.set_image_data_format

tf.keras.backend.set_image_data_format函数

tf.keras.backend.set_image_data_format(data_format)

定义在：tensorflow/python/keras/backend.py。

设置图像数据格式约定的值。

参数：

• data_format：字符串，'channels_first'或'channels_last'。

例：

>>> from keras import backend as K
>>> K.image_data_format()
'channels_first'
>>> K.set_image_data_format('channels_last')
>>> K.image_data_format()
'channels_last'

可能引发的异常：

• ValueError：如果data_format值无效。