如何将keras网络改为channels_last而不是channels_first？

Question

改变网络的代码是第一的还是最后的？我知道你可以把它放在图层本身，但是我每次都要写data_format='channels_last'吗？我有点记得有一个更容易的方法，但我已经忘记了，因此我的问题。

然而，这是一种做法：

model = tf.keras.Sequential([tf.keras.layers.Conv2D(1,(3,3),data_format="channels_last")])

Answer 1

对于版本tensorflow>=2.2.0，默认设置为"channels_first“(NCHW)。我们可以轻松地将"channels_last“改为”NHWC“

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

参考文献：https://www.tensorflow.org/api_docs/python/tf/keras/backend/set_image_data_format