tensorflow 模型加载（没有checkpoint文件或者说只加载其中一个模型）

最新推荐文章于 2023-01-30 12:53:23 发布

weixin_30376323

最新推荐文章于 2023-01-30 12:53:23 发布

阅读量2.2k

点赞数

CC 4.0 BY-SA版权

文章标签：人工智能运维 c#

原文链接：http://www.cnblogs.com/mengxiangtiankongfenwailan/p/10757665.html

本文详细介绍了在TensorFlow中如何加载和恢复模型，包括有checkpoint文件和仅有一个模型文件的情况。通过具体代码示例，展示了使用import_meta_graph和restore函数进行模型加载的过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1、如果有checkpoint文件的话，加载模型很简单：

　　第一步：都是加载图：

with tf.Session() as sess: 
　　saver=tf.train.import_meta_graph('./insightface_iter_best_71000.ckpt.meta')
　　saver.restore(sess,tf.train.latest_checkpoint('./'))

2、没有该文件，只有一个模型的时候：

with tf.Session() as sess: 
　　saver=tf.train.import_meta_graph('./insightface_iter_best_71000.ckpt.meta')
　　saver.restore(sess,'./insightface_iter_best_71000.ckpt')  # 重点，将地址写到.ckpt

转载于:https://www.cnblogs.com/mengxiangtiankongfenwailan/p/10757665.html