tensorflow 模型加载(没有checkpoint文件或者说只加载其中一个模型)
作者:互联网
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
标签:sess,模型,checkpoint,ckpt,meta,tf,saver,加载 来源: https://www.cnblogs.com/mengxiangtiankongfenwailan/p/10757665.html