Keras 模型中的数据格式
Data Format in Keras Model
我正在使用卷积网络,我想知道我的代码中使用了哪个 data_format
?因为从这个页面:Convolutional Layers
在 Conv2D 层部分,它表示输入形状可以采用两种不同的形式,具体取决于 data_format
("channels_first"
或 "channels_last"
)。我假设其中一个是默认选项?
我应该在我的代码中添加什么命令以在这两种格式之间切换?
哦,刚刚找到答案,在这里:
因此,如果使用 TensorFlow 后端到 keras,它使用 "channels_last"
,如果使用 Theano 后端,它使用 "channels_first"
。
我正在使用卷积网络,我想知道我的代码中使用了哪个 data_format
?因为从这个页面:Convolutional Layers
在 Conv2D 层部分,它表示输入形状可以采用两种不同的形式,具体取决于 data_format
("channels_first"
或 "channels_last"
)。我假设其中一个是默认选项?
我应该在我的代码中添加什么命令以在这两种格式之间切换?
哦,刚刚找到答案,在这里:
因此,如果使用 TensorFlow 后端到 keras,它使用 "channels_last"
,如果使用 Theano 后端,它使用 "channels_first"
。