TensorFlow 2 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: python 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.
|