Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tf.keras.backend.set_image_data_format

TensorFlow 1 version View source on GitHub

Sets the value of the image data format convention.

tf.keras.backend.set_image_data_format(
    data_format
)

Arguments:

  • data_format: string. 'channels_first' or 'channels_last'.

Example:

tf.keras.backend.image_data_format() 'channels_last' tf.keras.backend.set_image_data_format('channels_first') tf.keras.backend.image_data_format() 'channels_first' tf.keras.backend.set_image_data_format('channels_last')

Raises:

  • ValueError: In case of invalid data_format value.