我正在尝试使用预训练网络来提取新数据集的特征,例如 Google 在 tensorflow (inception-2015-12-05.tgz) 中发布的一个分类图像_graph_def.pb。我在这方面取得了成功,因为 transfer_learning 有教程,它使用分类图像_graph_def.pb(inception_v3.pb)来提取新数据集的裂缝。
然而,在新版本的预训练模型中,tensorflow 提供了检查点文件(例如 resnet_v1_152.ckpt)而不是 Graph_def(例如 resnet_v1_152.pb)。我想知道如何使用这些检查点文件来提取 transfer_learning 中的特征。谁能给我一些方向?
最佳答案
只需按照官方模型保存/恢复文档 here 即可。
关于python-2.7 - 预训练检查点 (.chkpt) 与 GraphDef (.pb),我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/39690130/