Не удается преобразовать файл модели в .tflite

Question

Я клонировал этот Image Completion репозиторий, и он работает на моем компьютере с Windows в Python.Размещены модели для лица-завершения и общего-завершения .Однако файлы не имеют расширения .pb.Как я могу получить название входных и выходных тензоров?Я попытался переименовать файлы с расширением .pb и попытался загрузить в тензорную доску, используя эту статью , но это дает ошибку, показанную на скриншоте.

Я также попытался запустить это в python, используя этот ответ, но я не получаю узлов на выходе.

Как я могу получить такие детали, как тензоры ввода и вывода, чтобы преобразовать эту модель в модель .tflite для Android?Снимки экрана:
ошибка импорта модели
ошибка тензорной доски

Answer 1

Файлы, представленные в хранилище, представляют собой контрольные точки , а не .pb файлы, которые можно преобразовать в .tflite.Это видно из вызова saver.restore(), сделанного в demo.py.

. Чтобы получить файл pb, вам нужно будет создать SavedModel .Полученная папка будет содержать .py файл , который можно использовать для преобразования (при условии, что он содержит операции с поддержкой TFLite).