4

我正在尝试通过使用预先训练的网络来提取新数据集的特征,例如谷歌在张量流(inception-2015-12-05.tgz)中发布的classify_image_graph_def.pb。我成功了,因为transfer_learning有教程,它使用分类图像图形定义.pb (inception_v3.pb) 来提取新数据集的裂缝。

但是,在新版本的预训练模型中,tensorflow 提供了检查点文件(例如 resnet_v1_152.ckpt)而不是 Graph_def(例如 resnet_v1_152.pb)。我想知道如何使用这些检查点文件来提取像transfer_learning中的特征。谁能给我一些指示?

4

1 回答 1

0

只需按照此处的官方模型保存/恢复文档即可。

于 2017-10-20T19:43:21.553 回答