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