在Python 2.7中加载Tensorflow模型

时间:2018-08-13 01:06:20

标签: python tensorflow

我有一个经过全面培训和保存的Tensorflow模型,我想加载该模型并将其用作第三方应用程序(UCSF Chimera)的插件。该第三方应用程序在不支持Tensorflow的Python 2.7上运行。如果可能,我是否有办法在python 2.7中完全使用此模型?

我原本是在看this的上一篇文章,但这是针对Java / C ++的。

1 个答案:

答案 0 :(得分:1)

首先,使用pickle

保存Tensorflow模型
with open("xxx.pkl", "wb") as outfile:
    pickle.dump(checkpointfile, outfile)

第二,安装anaconda并创建一个python2.7环境

第三,在python2.7环境中再次安装tensorflow

conda install tensorflow

第四,使用pickle

读取模型
pkl_file = open("xxx.pkl", "rb")
data = pickle.load(pkl_file, encoding="latin1")