方法1
在users文件夹下新建.keras文件夹,在文件夹下新建keras.json文件,json内容为:
{
"image_dim_ordering": "tf",
"epsilon": 1e-07,
"floatx": "float32",
"backend": "tensorflow"
}
这其中的backend可以是tensorflow或者theano,当然前提是你已经安装好了。
但是上面这种方式有一个弊端就是每次当你想切换backend的时候都需要手动修改这个文件,所以有一种能够动态修改的方式不是更好吗?请看下面的方法:
方法2
import os
os.environ['KERAS_BACKEND'] = 'tensorflow'
import keras.backend as K
K.set_image_dim_ordering('tf')
首先载入os库,将keras的backend修改为tensorflow,再将keras的backend加载,设置order为tensorflow格式。
本文介绍了两种配置Keras后端的方法:一种是通过创建特定的文件夹和配置文件来固定后端类型;另一种则是在代码中动态指定后端类型及图片维度顺序。
1万+

被折叠的 条评论
为什么被折叠?



