在图* meta文件之外加载tensorflow模型

时间:2018-11-06 18:23:31

标签: tensorflow

从其他一些问题(例如 this)中,我得出结论,在加载计算图时,通常需要三个文件:

 ckp.data-00000-of-00001
 ckp.index
 ckp.meta

其中*meta文件描述了所述图的结构。我正在尝试使用预训练的模型(供参考,here的MobileNetV2在城市景观上训练),但是提供的文件如下:

- a frozen inference graph (frozen_inference_graph.pb)
- a checkpoint (model.ckpt.data-00000-of-00001, model.ckpt.index)

我现在不想使用冻结的图,因为我需要能够微调模型。但是,如您从上面的描述中看到的,*meta文件丢失了,而且我无法在其存储库中找到用于生成图的相关代码(对于tensorflow来说,我可能一直在寻找错误的位置)。

在此处加载模型的最佳方法是什么?我是否应该加载冻结的图并从那里提取*meta文件中的信息?

0 个答案:

没有答案