C#TensorFlow接收模型(.ckpt)还原并生成冻结的模型文件(.pb)

C#TensorFlow接收模型(.ckpt)还原并生成冻结的模型文件(.pb),c#,tensorflow,tensorflow-slim,C#,Tensorflow,Tensorflow Slim,我想下载并恢复它,为C#tensorflow程序制作一个冻结的模型文件。(.PB) 参考: Python代码如下: import tensorflow as tf slim = tf.contrib.slim from inception_resnet_v2 import inception_resnet_v2, inception_resnet_v2_arg_scope height = 299 width = 299 channels = 3 X = tf.placeholder(tf

我想下载并恢复它,为C#tensorflow程序制作一个冻结的模型文件。(.PB)

参考:

Python代码如下:

import tensorflow as tf
slim = tf.contrib.slim

from inception_resnet_v2 import inception_resnet_v2, inception_resnet_v2_arg_scope

height = 299
width = 299
channels = 3

X = tf.placeholder(tf.float32, shape=[None, height, width, channels])
with slim.arg_scope(inception_resnet_v2_arg_scope()):
     logits, end_points = inception_resnet_v2(X, num_classes=1001,is_training=False)

saver = tf.train.Saver()
sess = tf.Session()
saver.restore(sess, "./inception_resnet_v2_2016_08_30.ckpt")

tf.train.write_graph( sess.graph_def, "./pb", 'inception_resnet_v2_2016_08_30.pb', as_text=True)

《开始》\u resnet\u v2\u 2016\u 08\u 30.ckpt[226MB]

初始版本(resnet)v2(u 2016)08(u 30.pb)[6MB]

输出文件不正确

如何获取完整的.pb文件

创建冻结的模型文件参照: github.com/migueldeicaza/TensorFlowSharp/issues/85#issuecomment-313590825

命令:

python .\Python\Python35\Lib\site-packages\tensorflow\python\tools\freeze_graph.py --   input_graph=.\inception_resnet_v2_2016_08_30.pb --input_checkpoint=.\inception_resnet_v2_2016_08_30.ckpt --output_node_names=Output/Predictions,Output/Loss --output_graph=inception_resnet_v2_2016_08_30_out.pb

错误:断言错误:输出/预测不在图形中

该错误意味着您的图形没有输出/预测。可以使用图中的节点名称吗?或者添加一个tf.identity(predictions,name=“Output/predictions”)错误表示图形没有输出/预测。可以使用图中的节点名称吗?或者添加一个,带有tf.identity(predictions,name=“Output/predictions”)