keras版本yolov3的后端切换问题

2020-07-01  本文已影响0人  darkness605

因为我是在服务器的虚拟环境下跑代码,所以修改home/user/下的keras.json文件并没有效果。

看网上大佬说是修改json文件可以切换keras后端的。

{ # keras.json文件内容

    "image_dim_ordering": "tf",

    "epsilon": 1e-07,

    "floatx": "float32",

    "backend": "theano"

}

就是修改backend这里的值。

以上方法都试过,没有作用,记录一下以防以后碰到同样的问题。

最后通过在python代码中import keras前加入一个环境变量修改的语句

import os

os.environ['KERAS_BACKEND']='tensorflow'

完美解决了。

上一篇下一篇

猜你喜欢

热点阅读