tf.keras.backend.set_image_data_format 

tf.keras.backend.set_image_data_format(data_format)

Defined in tensorflow/python/keras/backend.py.

Sets the value of the image data format convention.

Arguments:

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

Example:

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

Raises:

  • ValueError: In case of invalid data_format value.