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

 


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM