tf.keras.backend.set_image_data_format
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. |
