其他分享
首页 > 其他分享> > tensorflow 模型加载(没有checkpoint文件或者说只加载其中一个模型)

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