在Tensorflow 1.2中将经过训练的模型从Python加载到C ++

时间:2017-07-10 12:21:24

标签: python c++ tensorflow

我正在寻找使用Tensorflow的C ++ API v1.2为MNIST数据集运行基本的全连接神经网络。我已经训练了模型并使用Python中的tf.train.Saver()导出它。这给了我一个检查点文件,一个数据文件,一个索引文件和一个元文件。

我知道数据文件包含已保存的变量,而元文件包含在先前项目中使用Tensorboard的图表。

  

但是,我不确定加载这些文件的推荐方法是什么   并且在v1.2中的C ++环境中运行训练过的模型,因为所有的   我发现的教程和问题适用于不同的旧版本   基本上

我发现tensorflow::ops::Restore应该是执行此类操作的方法,但我知道Tensorflow中的推理不受支持,因此我不确定应该按顺序给出什么参数接收训练有素的模型,我可以将其放入session->Run()并在输入测试数据时收到准确性声明。

0 个答案:

没有答案