加载预训练的 resnet 模型时出错

Question

我正在尝试在下面加载预训练的 ResNet 模型 link https://drive.google.com/open?id=1xkVK92XLZOgYlpaRpG_-WP0Elzg4ewpw

但它给出了 RuntimeError: The Session graph is empty。在调用运行().

之前向图中添加操作

可能是什么问题？

import tensorflow as tf
import tensorflow.contrib.slim as slim

# Let's load a previously saved meta graph in the default graph
# This function returns a Saver
saver = tf.train.import_meta_graph('model.ckpt-0.meta')

# We can now access the default graph where all our metadata has been loaded
graph = tf.get_default_graph()

with tf.Session(graph=tf.Graph()) as sess:

    saver.restore(sess, 'model.ckpt-0.data-00000-of-00001')
    print('Worked')

Answer 1

你必须有一个模型（粗糙的房子），和负载参数（床，家具）。现在你需要一个粗糙的房子（操作，例如as:tf。变量（），tf.add（ ),tf.nn.softmax_cross_entropy_with_logits()).

Answer 2

with tf.Session() as sess:
#  tf.saved_model.loader.load(sess, [tag_constants.TRAINING], export_dir)

  saver = tf.train.import_meta_graph('C://Users//hardi//tutorial//resnet//model.ckpt.meta')
#  new_saver   = saver.restore(sess, tf.train.latest_checkpoint('C://Users//hardi//tutorial//resnet//'))
  saver.restore(sess, 'model.ckpt')
  graph = tf.get_default_graph()
  print('success')

错误是将保护程序实例带入循环并使用 'model.ckpt' 而不是 'model.ckpt-0.data-00000-of-00001' 作为 V2 检查点在这里找到解决方案 https://github.com/tensorflow/models/issues/2676

加载预训练的 resnet 模型时出错

Error while loading a pretrained resnet model

tensorflow

resnet

pre-trained-model