TensorFlow函數(shù)教程：tf.keras.backend.set_image_data_format

tf.keras.backend.set_image_data_format函數(shù)

tf.keras.backend.set_image_data_format(data_format)

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

設置圖像數(shù)據(jù)格式約定的值。

參數(shù)：

• 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'

可能引發(fā)的異常：

• ValueError：如果data_format值無效。