abstractnn.onnx_parser

Parser pour modèles ONNX

Classes

ONNXParser(model_path)

Parse et extrait les opérations d'un modèle ONNX

class abstractnn.onnx_parser.ONNXParser(model_path: str)

Parse et extrait les opérations d’un modèle ONNX

__init__(model_path: str)

parse() → List[Dict[str, Any]]

Parse le graphe ONNX et extrait les couches

get_input_shape() → tuple

Retourne la forme de l’entrée du modèle

get_output_names() → List[str]

Retourne les noms des sorties