Ошибка при загрузке предварительно обученной модели повторной сети - PullRequest
0 голосов
/ 08 июля 2019

Я пытаюсь загрузить предварительно обученную модель ResNet по ссылке ниже https://drive.google.com/open?id=1xkVK92XLZOgYlpaRpG_-WP0Elzg4ewpw

Но она выдает RuntimeError: График сеанса пуст.Добавьте операции на график перед вызовом run ().

В чем может быть проблема?

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')

Ответы [ 2 ]

0 голосов
/ 09 июля 2019
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

0 голосов
/ 08 июля 2019

у вас должны быть модель (грубый дом) и параметр загрузки (кровать, мебель). Теперь вам нужен грубый дом (операции, такие как: tf.Variable (), tf.add (), tf.nn. softmax_cross_entropy_with_logits ()).

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...