Сохраните файлы .pb и .pbtxt из keras

Мне нужно получить файлы .pb и .pbtxt из модели Keras, чтобы использовать обученный Модель Keras с модулем DNN OpenCV на C++.

Я могу получить файл .pb, выполнив model.save('model',save_format='tf'), но я не нашел способа получить файл .pbtxt.


person Barabanga    schedule 09.06.2020    source источник


Ответы (1)


Я работаю с той же настройкой и у меня были те же проблемы, поэтому вот что я нашел:

  1. В OpenCV, если вы используете dnn::readNetFromTensorflow(), вам на самом деле не нужен файл «.pbtxt», достаточно одного «.pb».

  2. То, что вы создаете с помощью метода keras model.save(), представляет собой тензорный поток SavedModel, который генерирует несколько файлов (вместе с .pb), но .pb не работает в одиночку, и OpenCV не принимает формат SavedModel. Итак, вам нужно создать замороженный график. Дополнительная информация здесь, а также решение для создания замороженного графика для вашей модели.

person Coucou436    schedule 13.06.2020