Python 3: Tensorflow Как загрузить мою модель через ноутбук Jupyter - PullRequest
Новая
       36

Python 3: Tensorflow Как загрузить мою модель через ноутбук Jupyter

0 голосов
/ 25 апреля 2020

Ι использовал предварительно обученную модель, а я обучил свою, поэтому мне нужно обслуживать мою модель из-за функции load (). Вот код из предварительно обученной модели, откуда она загружается из Интернета. Моя модель сохраняется на диске в каталоге TensorFlow ~ \ tenorflow \ models \ research \ object-Detection. 

def load_model(model_name):
  base_url = 'http://download.tensorflow.org/models/object_detection/'
  model_file = model_name + '.tar.gz'
  model_dir = tf.keras.utils.get_file(
    fname=model_name, 
    origin=base_url + model_file,
    untar=True)

  model_dir = pathlib.Path(model_dir)/"saved_model"

  model = tf.saved_model.load(str(model_dir))
  model = model.signatures['serving_default']

  return model

Я пытаюсь использовать другую функцию tf.keras tf.keras.models.load_model 

def load_model(model_name):
  #base_url = 'http://download.tensorflow.org/models/object_detection/'
  model_file = model_name
  model_dir = tf.keras.models.load_model(filepath='/research/object_detection/cellphone_inference_graph',
custom_objects=None, compile=True)

  model_dir = pathlib.Path(model_dir)/"saved_model"

  model = tf.saved_model.load(str(model_dir))
  model = model.signatures['serving_default']

  return model

Но когда я запускаю свой проект, все равно появляется ошибка 

OSError: SavedModel file does not exist at: /research/object_detection/cellphone_inference_graph/{saved_model.pbtxt|saved_model.pb

Кто-нибудь знает, как загрузить мою модель? Или любую ссылку, чтобы прочитать, как это сделать, потому что в Интернете я нашел только конфигурацию самой старой версии object_detection_tutorial.ipynb

Спасибо U

...