如何从checkpoint目录冻结初始-v3模型的图形?

时间:2017-03-15 21:11:03

标签: tensorflow

我正在使用这个来调整inception-v3模型花:https://github.com/tensorflow/models/tree/master/inception

我在一个目录中检查了结果。但在目录中我看到这样的文件: -rw-r--r-- 1 root root 389908432 Mar 15 21:46 model.ckpt-0.data-00000-of-00001 -rw-r--r-- 1 root root 72680 Mar 15 21:46 model.ckpt-0.index -rw-r--r-- 1 root root 15189794 Mar 15 21:47 model.ckpt-0.meta -rw-r--r-- 1 root root 135185788 Mar 15 22:36 events.out.tfevents.1489594533.f7d5defbed64 -rw-r--r-- 1 root root 72680 Mar 15 22:37 model.ckpt-4999.index -rw-r--r-- 1 root root 389908432 Mar 15 22:37 model.ckpt-4999.data-00000-of-00001 -rw-r--r-- 1 root root 15189794 Mar 15 22:38 model.ckpt-4999.meta -rw-r--r-- 1 root root 130 Mar 15 22:49 checkpoint 而我需要目录中的输出类似于: -rw-r----- 1 107456 5000 223 Mar 2 2016 README.txt -rw-r----- 1 107456 5000 43 Mar 2 2016 checkpoint -rw-r----- 1 107456 5000 434903494 Mar 15 2016 model.ckpt-157585

为此,我需要做一些像冻结的事情,但冻结需要提供output_node_names。任何人都可以指导我,初始-v3的output_node_names是什么? 另外,我需要一种可靠的冻结方式。 tensorflow冷冻工具可以吗?

1 个答案:

答案 0 :(得分:0)

我最终找到了答案。

单行答案应该是使用上游Tensorflow代码库中的freezer.py。请参阅有关如何从测试中使用该程序的示例。

您可以查看以下链接以获取样本: https://gist.githubusercontent.com/morgangiraud/249505f540a5e53a48b0c1a869d370bf/raw/6cb0b4d497925517316a92f935ce5dccb6aafd17/medium-tffreeze-1.py