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

佚名 6年前 (2019-04-23) 随笔 1381人围观抢沙发百度已收录

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

　　第一步：都是加载图：

SRE实战互联网时代守护先锋，助力企业售后服务体系运筹帷幄！一键直达领取阿里云限量特价优惠。

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