python-2.7 – 预先训练的检查点(.chkpt)与GraphDef(.pb)

概述我试图通过使用预先训练的网络来提取新数据集的功能,例如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. 总结

以上是内存溢出为你收集整理的python-2.7 – 预先训练的检查点(.chkpt)与GraphDef(.pb)全部内容，希望文章能够帮你解决python-2.7 – 预先训练的检查点(.chkpt)与GraphDef(.pb)所遇到的程序开发问题。

如果觉得内存溢出网站内容还不错，欢迎将内存溢出网站推荐给程序员好友。