可以只用 .ckpt-meta 和恢复一个模型 .ckpt-data 已知ckpt模型转pb模型 1.查看输出节点名称 # function: get the node name of ckpt model from tensorflow.python import pywrap_tensorflow # checkpoint_path = 'model.ckpt-xxx' #保留ckpt的文件目录 checkpoint_path = './DnCNN_tensorflow1_4/checkpoint_dem...
# 方式一:加载指定文件夹下最近保存的一个模型的数据 saver.restore(sess, tf.train.latest_checkpoint('./')) # 方式二:指定具体某个数据,需要注意的是,指定的文件不要包含后缀 # saver.restore(sess, os.path.join(path, 'model.ckpt-1000')) # 查看模型中的trainable variables tvs = [v for v in ...