Tensoflow 2.0 C ++ - загрузить предварительно обученную модель

Question

Может ли кто-нибудь дать мне подсказку, как я могу загрузить модель, обученную и экспортированную в python с помощью keras, с помощью C ++ API тензор потока 2.0?

Я не могу найти информацию об этом, только с тензорной версией <2. </p>

С уважением

Answer 1

Хорошо, я нашел решение гайки с другими проблемами:

В Python вы должны экспортировать его с:

tf.keras.models.save_model(model, 'model')

В C ++ вы должны загрузить его с:

tensorflow::SavedModelBundle model;
tensorflow::Status status = tensorflow::LoadSavedModel(
  tensorflow::SessionOptions(), 
  tensorflow::RunOptions(), 
  "path/to/model/folder", 
  {tensorflow::kSavedModelTagServe}, 
  &model);

Основано на этом посте: Использование контрольной точки Tensorflow для восстановления модели в C ++

Если я сейчас пытаюсь установить входы и выходы, выдается ошибка: «Не удалось найти узел с именем 'outputlayer' "и" Неверный аргумент: вход Тензор: 0, указанный либо в feed_devices, либо в fetch_devices не был в Графике ".

У кого-нибудь есть идея, что здесь не так?