美文网首页
keras版本yolov3的后端切换问题

keras版本yolov3的后端切换问题

作者: darkness605 | 来源:发表于2020-07-01 18:50 被阅读0次

    因为我是在服务器的虚拟环境下跑代码,所以修改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'

    完美解决了。

    相关文章

      网友评论

          本文标题:keras版本yolov3的后端切换问题

          本文链接:https://www.haomeiwen.com/subject/ndzrqktx.html