关于 keras 中的 set_image_dim_ordering('tf')

Question

在尝试 Keras 实现时，我执行了以下操作：

from keras import backend as K
K.set_image_dim_ordering('tf') 

它会生成以下错误消息。这是什么意思？我认为 set_image_dim_ordering 包含在 Keras 中。

File "train.py", line 14, in <module>
K.set_image_dim_ordering('tf')

AttributeError：“模块”对象没有属性“set_image_dim_ordering”

Answer 1

Keras 是深度学习框架，可以使用“ TensorFlow ”、“ Theano ”和“ CNTK ”作为后端。每个后端都有自己的使用通道排序的偏好

• Theano：通道一阶
• Tensorflow：通道最后顺序
• CNTK : 频道最后订单

如果您使用“ Theano ”作为后端，您应该设置“ channels_first order”，并且在导入 keras 库后您可以使用这行代码：

keras.backend.set_image_data_format('channels_first')

数据形状如下：（通道、行、列）。

注意： “数据形状”中的通道顺序应等于“后端”中的通道顺序。例如，如果您使用Tensorflow作为后端，那么数据的输入形状应该是通道最后顺序（对于 RGB 图像：（行，列，通道））