python-2.7 – 预先训练的检查点(.chkpt)与GraphDef(.pb)
发布时间:2020-12-20 13:14:11 所属栏目:Python 来源:网络整理
导读:我试图通过使用预先训练的网络来提取新数据集的功能,例如Google在tensorflow(inception-2015-12-05.tgz)中发布的classify_image_graph_def.pb.我之所以成功,因为 transfer_learning上有教程,它使用classify_image_graph_def.pb(inception_v3.pb)来提取新数据
我试图通过使用预先训练的网络来提取新数据集的功能,例如Google在tensorflow(inception-2015-12-05.tgz)中发布的classify_image_graph_def.pb.我之所以成功,因为
transfer_learning上有教程,它使用classify_image_graph_def.pb(inception_v3.pb)来提取新数据集的碎片.
但是,在新发布的预训练模型中,tensorflow提供了检查点文件(例如resnet_v1_152.ckpt)而不是Graph_def(例如resnet_v1_152.pb).我想知道如何使用这些检查点文件来提取功能,如transfer_learning.有人能给我一些指示吗? 解决方法
只需按照官方模型保存/恢复文档
here.
(编辑:李大同) 【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容! |