Scripts: Informações sobre entradas e saídas do modelo ONNX

 

Informações sobre entradas e saídas do modelo ONNX:

Script para obter informações sobre o número, tipos e tamanhos de tensores de entrada e saída do modelo onnx

Informações sobre entradas e saídas do modelo ONNX

Autor: Slava

 
Automated-Trading:
Script para obter informações sobre o número, tipos e tamanhos de tensores de entrada e saída do modelo onnx

exemplo de código em Python usando a biblioteca onnx para obter informações sobre os tensores de entrada e saída de um modelo ONNX:

Certifique-se de substituir "caminho/do/modelo.onnx" pelo caminho do arquivo ONNX do seu modelo. Esse código carregará o modelo ONNX, percorrerá os tensores de entrada e saída e exibirá seus nomes, tipos e formatos.

Espero que possa ter ajudado

 

falta o suporte?)

 
floston #:

falta o suporte?)

Não. O formato foi alterado. O script atualizado está sendo verificado pelo moderador.

Aqui, experimente o script atualizado com o modelo de cluster iris (esse é o motivo da alteração). Altere o nome para iris.onnx

Arquivos anexados:
iris.txt  5 kb