Есть ли способ восстановить файл .pb или .meta из файлов .index и .ckpt? - PullRequest
0 голосов
/ 19 июня 2020

Я загрузил код с github, который содержит файл .index и файл контрольной точки (сеть обучается с использованием tensorflow 2.1 и python 3.7). Но он не содержит файлов .meta или .pb. Есть ли способ восстановить файл .pb или .meta из файла .index и файла контрольной точки?

1 Ответ

1 голос
/ 28 июня 2020

Это может быть очень конкретным c, но если вы пометили объект-обнаружение-api, это может сработать:

Вы можете проверить export_inference_graph.py из object_detection API.

С примером вызова, подобным этому:

python export_inference_graph.py \
--input_type image_tensor \
--pipeline_config_path path/to/ssd_inception_v2.config \
--trained_checkpoint_prefix path/to/model.ckpt \
--output_directory path/to/exported_model_directory

Это будет работать, если у вас есть модель, основанная на api обнаружения объектов и уже обученная там. Так, например, если вы нашли модель в model Zoo , вы можете найти конфигурации в здесь и использовать их в качестве файла .config в приведенном выше вызове.

Если вы используете модель tf lite, вы можете посмотреть export_tflite_ssd_graph.py

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