Скрипты: Информация о входах и выходах onnx-модели

 

Информация о входах и выходах onnx-модели:

Скрипт для получения информации о количестве, типах и размерах входных и выходных тензоров onnx-модели

Автор: Slava

 
Automated-Trading:
Скрипт для получения информации о количестве, типах и размерах входных и выходных тензоров модели onnx

Пример кода на языке Python с использованием библиотеки onnx для получения информации о входных и выходных тензорах модели ONNX:

Убедитесь, что вы заменили "path/of/model.onnx" на путь к ONNX-файлу вашей модели. Этот код загрузит модель ONNX, переберет входные и выходные тензоры и отобразит их имена, типы и форматы.

Надеюсь, это помогло.

 

скобку пропустили?)

 
floston #:

скобку пропустили?)

Нет. Формат поменялся. Обновлённый скрипт на проверке у модератора.

Вот, попробуйте обновлённый скрипт с кластерной моделью iris (из-за неё и меняли). Поменяйте имя на iris.onnx

Файлы:
iris.txt  5 kb