我尝试使用freeze_graph.py脚本冻结graph.pbtxt和model.ckpt-2000,但是在下面出现错误。
TypeError:names_to_saveables必须是将字符串名称映射到张量/变量的字典。不是变量:Tensor(“ dnn / hiddenlayer_0 / bias:0”,shape =(10,),dtype = float32)
这是我使用的命令。
python Frozen_graph.py --input_graph = graph.pbtxt --input_checkpoint = model.ckpt-2000 --output_graph = frozen_graph.pb --output_node_names = output_node --input_binary = false
该模型来自此页面中的示例。 https://www.tensorflow.org/versions/r1.3/get_started/estimator
feature_columns = [tf.feature_column.numeric_column(“ x”,shape = [4])]
classifier = tf.estimator.DNNClassifier(feature_columns = feature_columns,hidden_units = [10,20,10],n_classes = 3,model_dir =“ / tmp / iris_model”)
知道为什么它不起作用吗?