Невозможно преобразовать .pb в .nn с помощью tensorflow_to_barracuda.py - PullRequest
0 голосов
/ 08 июля 2020

Я пытаюсь преобразовать специально обученную модель тензорного потока в ssd_mobilenet_v1_coco в формат .nn для использования в Unity. Я пробовал код tensorflow_to_barracuda.py как с файлами frozen_inference_graph.pb, так и с сохраненными_model.pb, и это ошибки, которые я получил. Я искал решение, но не нашел. Помогите мне. 1010 *

Активированная среда tenorflow Клонированное репозиторий barracuda из https://github.com/Unity-Technologies/barracuda-release Попытка преобразовать файл модели TF с помощью tensorflow_to_barracuda.py Используемая команда : python barracuda-release-release-1.0.0 / Tools / tensorflow_to_barracuda.py saved_model / saved_model saved_model / saved_model.nn

Версия Unity: 2019.4.1f1 Версия ОС +: Windows 10 Версия ML-Agents: 1.0.2 Версия TensorFlow: 1.14

...