使用TF slim冻结resnet_v1_50的图形失败

Question

我想冻结模型并部署它们。我首先使用InceptionV3遵循了slim从1到2附带的教程，效果很好。然后我尝试对resnet_v1_50做同样的操作，它是从提供的url下载的。

如有任何建议，我们将不胜感激！

谢谢！

以下是步骤：

cd /path/to/tensorflow-models/research/slim
# export inference graph, success
python export_inference_graph.py --alsologtostderr --model_name=resnet_v1_50 --output_file=resnet_v1_50_inf_graph.pb
# freeze the inference graph with the trained weights, failed
python ~/.local/lib/python2.7/site-packages/tensorflow/python/tools/freeze_graph.py 
--input_graph=resnet_v1_50_inf_graph.pb 
--input_checkpoint=/path/to/models/resnet_v1_50.ckpt 
--input_binary=true --output_graph=frozen_resnet_v1_50.pb 
--output_node_names=resnet_v1_50/predictions/Reshape

错误消息太长，我重定向了一条消息，将其引用到此处。

Answer 1

如下行所示，导出推理图时应将labels_offset设置为1。

InvalidArgumentError (see above for traceback): Assign requires shapes of both 
tensors to match. lhs shape= [1001] rhs shape= [1000]
[[Node: save/Assign_265 = Assign[T=DT_FLOAT, _class= . 
["loc:@resnet_v1_50/logits/biases"], use_locking=true, validate_shape=true, 
_device="/job:localhost/replica:0/task:0/device:GPU:0"](resnet_v1_50/logits/biases, 
save/RestoreV2_265/_1)]]

命令将类似于：

python export_inference_graph.py --model_name=resnet_v1_50 --labels_offset=1 --output_file=resnet_v1_50_inf_graph.pb