一、修改backend的方法
(一)找到.keras
文件夹,在文件夹下新建keras.json
文件,json内容为:
{ "image_dim_ordering": "tf", "epsilon": 1e-07, "floatx": "float32", "backend": "tensorflow" }
这种方式有一个弊端就是每次当你想切换backend的时候都需要手动修改这个文件
(二)动态修改
import os os.environ['KERAS_BACKEND'] = 'tensorflow' import keras.backend as K K.set_image_dim_ordering('tf')
二、backend函数
见https://www.cnblogs.com/aibbtcom/p/8541794.html