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

 

posted @   青牛梦旅行  阅读(3435)  评论(0编辑  收藏  举报
编辑推荐:
· 开发者必知的日志记录最佳实践
· SQL Server 2025 AI相关能力初探
· Linux系列:如何用 C#调用 C方法造成内存泄露
· AI与.NET技术实操系列(二):开始使用ML.NET
· 记一次.NET内存居高不下排查解决与启示
阅读排行:
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· 没有Manus邀请码?试试免邀请码的MGX或者开源的OpenManus吧
· 【自荐】一款简洁、开源的在线白板工具 Drawnix
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY
· Docker 太简单,K8s 太复杂?w7panel 让容器管理更轻松!
点击右上角即可分享
微信分享提示