利用tensorflow的upgrade.py将之前的代码自动升级为1.0版本的代码

来源:互联网 发布:休闲食品网络市场调查 编辑:程序博客网 时间:2024/05/22 04:50

    前几天在win10上装了tensorflow1.0,是最新版本的,不得不说,对于有python基础的认来说,学习tensorflow会相比caffe稍微轻松一些。在网上找资料学习的过程中,发现网上大部分的代码都是基于1.0之前的版本的,不能直接运行于1.0版本,因为1.0版本对很多接口的名称以及参数等都做了一些修改,如果不修改之前的代码是无法在1.0版本上运行的。

    修改之前的代码有两种方式,一种是查看tensorflow1.0版本修改的内容,手动对代码进行修改。

修改的内容可参照这个博客,其中对1.0版本修改的地方做了介绍,本人在此补充一点:1.0版本中将tf.merge_all_summaries重命名为tf.summary.merge_all。

    另一种方法是利用官方给的升级代码对1.0之前的代码进行升级,github中的代码地址。我下载好该代码进行升级时会报错,我电脑是win10 64为系统,不知是不是系统的原因。因此我对代码进行了修改,其实也只是在第520行添加了一句代码,修改后的文件放在了我的资源里,可点击下载。

    对代码进行修改后可以输出结果,但是输出后的文件在win10下直接运行会报错,将文件里的内容复制到新建的py文件中即可运行。

    下载好升级所需的py文件后,可在cmd中输入类似下面的语句升级代码:

    

python C:\Users\MaMQ\Documents\DeepLearning\tensorflow\tf_upgrade.py --infile C:\Users\MaMQ\Documents\DeepLearning\tensorflow\try\test_tensorboard.py --outfile C:\Users\MaMQ\Documents\DeepLearning\tensorflow\test.py

    其中,infile指需要转换的文件,outfile指转换后的文件。

    通过如上方法即可在win10下顺利升级以前的代码,好运。

0 1
原创粉丝点击