tensorflow: ckpt模型转成pb模型

---

title: ckpt模型转pb模型手册
tags: tensorflow,ckpt,pb

为什么要把ckpt格式转为pb格式？

ckpt格式的模型图结构和权重参数模型是分开保存在不同的文件里，而pb模型的图结构和权重参数的模型是保存在同一个文件中的,很多时候我们需要一个既包含图又包含变量的模型文件

ckpt格式模型包含的文件如下：

我们在checkpoint_dir目录下保存的文件结构如下：

1 meta文件保存的是图结构，meta文件是pb（protocol buffer）格式文件，包含变量、op、集合等。
2 checkpoint_dir目录下还有checkpoint文件，该文件是个文本文件，里面记录了保存的最新checkpoint文件以及其它checkpoint文件列表。在inference时，可以通过修改这个文件，指定使用哪个model。
3 data-00000-of-00001文件和.index文件就是ckpt文件，.data里面的内容存储的就是权重、偏置等内容。在TensorFlow0.11之前，使用ckpt一个后缀文件存储，以后的TensorFlow版本都是使用这两个文件共同存储模型参数。
4 index文件是一个不可变得字符串表，每一个键都是张量的名称，它的值是一个序列化的BundleEntryProto。 每个BundleEntryProto描述张量的元数据：“数据”文件中的哪个文件包含张量的内容ÿ