从其他一些问题(例如 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
文件中的信息?