Преобразование тензорного потока в ONNX
В настоящее время я пытаюсь преобразовать сохраненный (и работающий) файл.pb, созданный с помощью этого руководства ( https://github.com/thtrieu/darkflow), в файл onnx. В настоящее время я использую инструменты WinML, но результат преобразования не работает вообще (входные параметры неверны + вся архитектура неверна) .
Мой входной параметр (как указано в самом низу файла readme): input:0
Выходной узел: выход: 0
Я хочу использовать конвертированную модель в приложении UWP, которое работает на HoloLens.
Кто-нибудь уже успешно конвертировал эту модель (или любую модель TensorFlow) в ONNX? Если это так, я был бы очень благодарен за любую ссылку / учебник / помощь!
Пожалуйста, дайте мне знать, если вам нужна дополнительная информация. Я очень неопытен с такими форумами. Извини за это.
Вальтер
2 ответа
Вы можете использовать tf2onnx для преобразования файла .pb в onnx.
Установите его с помощью pip.
pip install tf2onnx
Тогда вы называете это так.
python -m tf2onnx.convert --input <path to your pb file> --inputs <comma-delimited input names goes here> --outputs <comma-delimited output names goes here> --output <path to output in onnx format>
Вы сказали, что ваш вклад
input:0и ваш вывод
ouput:0, поэтому команда должна выглядеть так.
python -m tf2onnx.convert --input yolo.pb --inputs input:0 --outputs ouput:0 --output yolo.onnx
При необходимости вы можете использовать приложение netron для проверки имени входа и выхода.