Tensorflow支持哪些类型的模型文件以及如何加载.ckpt + .pb

时间:2019-09-16 02:54:06

标签: tensorflow

我发现TensorFlow支持各种模型文件.ckpt, .meta, .pb

1)现在我有一个resnet50模型,.ckpt 102MB和.pb 459KB,如何使用TensorFlow将它们加载到内存中?

2)Tensorflow支持多少种模型文件?谁能提供类似的清单或材料(例如1. ckpt + pb 2. Frozen.pb 3. meta + ??)

1 个答案:

答案 0 :(得分:1)

这些文件不是用于存储完整模型的不同文件类型。每个模型都包含不同的信息,并且它们共同存储一个模型。

.meta文件存储模型的大部分结构,而.ckpt文件是检查点文件,用于存储有关网络内变量值(神经元的权重和偏差)的信息网络)处于保存网络的训练阶段。 .pb文件是协议缓冲区文件。

要查看如何使用这些文件保存和还原模型的工作示例,请参见此处:

https://bitbucket.org/FinleyGibson/save-and-load-tf-model/

并使用jupyter笔记本文件查看器打开.ipynb文件