Как экспортировать формат Keras h5 в TensorFlow .meta? - PullRequest
0 голосов
/ 02 мая 2018

Я использовал Keras, чтобы построить модель и обучил ее. Затем я сохранил модель в виде файла h5, т.е. model.save('name.h5'). Теперь я хочу перезагрузить модель в тензорном потоке, чтобы у меня был доступ к файлу .meta, например, я хочу импортировать вычислительный граф из файла .meta, то есть tf.train.import_meta_graph('name_of_the_file.meta').

Итак, вопрос в том, как преобразовать .h5 файл Keras в следующие четыре файла TensorFlow:

  1. .meta
  2. checkpoint
  3. .data-00000-of-00001
  4. .index

1 Ответ

0 голосов
/ 02 мая 2018

Вы можете использовать сторонние пакеты, например keras_to_tensorflow

keras_to_tensorflow: Общий код для преобразования обученной модели керас в модель тензорного потока вывода

Преобразование может быть сделано

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