加入收藏 | 设为首页 | 会员中心 | 我要投稿 李大同 (https://www.lidatong.com.cn/)- 科技、建站、经验、云计算、5G、大数据,站长网!
当前位置: 首页 > 编程开发 > Python > 正文

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.

(编辑:李大同)

【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容!

    推荐文章
      热点阅读