修改keras的后端。
方法参考Keras中文文档:http://keras-cn.readthedocs.io/en/latest/backend/
切换后端
注意:Windows用户请把$Home
改为%USERPROFILE%
如果你至少运行过一次Keras,你将在下面的目录下找到Keras的配置文件:
$HOME/.keras/keras.json
如果该目录下没有该文件,你可以手动创建一个
文件的默认配置如下:
{
"image_data_format": "channels_last",
"epsilon": 1e-07,
"floatx": "float32",
"backend": "tensorflow"
}
将backend
字段的值改写为你需要使用的后端:theano
或tensorflow
或者CNTK
,即可完成后端的切换
我们也可以通过定义环境变量KERAS_BACKEND
来覆盖上面配置文件中定义的后端:
KERAS_BACKEND=tensorflow python -c "from keras import backend;"
Using TensorFlow backend.
keras.json 细节
{
"image_data_format": "channels_last",
"epsilon": 1e-07,
"floatx": "float32",
"backend": "tensorflow"
}
你可以更改以上~/.keras/keras.json
中的配置
iamge_data_format
:字符串,"channels_last"或"channels_first",该选项指定了Keras将要使用的维度顺序,可通过keras.backend.image_data_format()
来获取当前的维度顺序。对2D数据来说,"channels_last"假定维度顺序为(rows,cols,channels)而"channels_first"假定维度顺序为(channels, rows, cols)。对3D数据而言,"channels_last"假定(conv_dim1, conv_dim2, conv_dim3, channels),"channels_first"则是(channels, conv_dim1, conv_dim2, conv_dim3)epsilon
:浮点数,防止除0错误的小数字floatx
:字符串,"float16"
,"float32"
,"float64"
之一,为浮点数精度backend
:字符串,所使用的后端,为"tensorflow"或"theano"