Voici les différents modules que propose l'API de ONNX :
| Fonction | Description |
|---|---|
| onnx.backend | Ce module permet d'exécuter des modèles ONNX sur différents backends. Il fournit une interface pour intégrer des backends afin de tester et exécuter des modèles sur différents cadre d'applications ou moteurs de calcul comme TensorFlow, PyTorch,... |
| onnx.checker | Ce module est utilisé pour vérifier la validité d'un modèle ONNX. Il permet de s'assurer que le modèle respecte les spécifications de l'API ONNX (comme la conformité des noeuds, la structure du graphe,...) avant de l'utiliser ou de le partager. |
| onnx.compose | Ce module permet de proposer des outils pour composer ou décomposer des modèles ONNX. Il peut être utilisé pour assembler des sous-modèles ou des blocs en un modèle plus large ou pour diviser un modèle en composantes plus petits. |
| onnx._custom_element_types | Ce module permet de gérer des types d'éléments personnalisés utilisés dans ONNX. Il permet de définir et gérer des types de données supplémentaires pouvant être utilisés dans certains modèles spécifiques. |
| onnx.defs | Ce module contient les définitions d'opérateurs. Ce module maintient la description des opérateurs (fonctions de traitement de base comme convolutions, addition,...) utilisés dans les modèles ONNX. |
| onnx.external_data_helper | Ce module permet de gérer les données externes associées aux modèles ONNX. Lorsqu'un modèle est trop grand, les poids et autres données peuvent être stockés dans des fichiers externes. Ce module aide à les lire et à les écrire. |
| onnx.helper | Ce module permet de fournir des fonctions utilitaires pour faciliter la création et la manipulation des modèles ONNX, comme la création de graphes, de noeuds, et l'édition des attributs. |
| onnx.hub | Ce module permet d'accéder à un dépôt de modèles ONNX pré-entraînés. Il aide à télécharger et utiliser ces modèles sans avoir à les construire depuis zéro. |
| onnx.inliner | Ce module permet de développer ou d'incorporer les sous-graphes directement dans le graphe principal. Cela simplifie l'architecture d'un modèle en incorporant les fonctions externes ou modules dans un seul graphe. |
| onnx.mapping | Ce module permet de gérer les cartographies de types de données entre les différents cadre d'applications interagissant avec ONNX. Il convertit les types de données d'autres cadre d'application en types ONNX et vice-versa. |
| onnx.model_container | Ce module permet de gérer la structure de conteneur de modèle ONNX. Il est utilisé pour manipuler les modèles (fichiers .onnx) et leur contenu, facilitant la sérialisation et désérialisation des modèles. |
| onnx.numpy_helper | Ce module contient des fonctions utilitaires pour interagir avec les tableaux NumPy. Il permet la conversion entre les types de données ONNX (comme les tenseurs) et les tableaux NumPy pour une meilleure compatibilité avec les outils d'analyse scientifique en Python. |
| onnx.parser | Ce module permet d'analyser et d'interpréter des fichiers ONNX en code ou en modèles interprétables. Il permet de lire des fichiers modèles dans une représentation plus compréhensible par les développeurs. |
| onnx.printer | Ce module permet de visualiser et d'afficher les structures des modèles ONNX dans un format lisible. Il est utile pour inspecter et déboguer la structure interne d'un modèle. |
| onnx.reference | Ce module contient une implémentation de référence pour l'exécution des modèles ONNX. Cela permet de vérifier le comportement d'un modèle sans avoir besoin de dépendre d'un backend particulier. |
| onnx.shape_inference | Ce module permet d'inférer les dimensions des tenseurs dans les modèles ONNX. Il permet de déterminer les formes des entrées et sorties des nouds en fonction des informations disponibles dans le graphe. |
| onnx.tools | Ce module permet de regrouper des outils supplémentaires pour manipuler les modèles ONNX, tels que l'optimisation des modèles ou la conversion de modèles à différentes versions. |
| onnx.utils | Ce module contient diverses fonctions utilitaires pour faciliter le traitement des modèles ONNX, comme l'édition des graphes, la vérification des propriétés des noeuds, ou la gestion des attributs des modèles. |
| onnx.version_converter | Ce module permet de convertir des modèles ONNX d'une version à une autre. Il est essentiel lorsque vous travaillez avec différentes versions de la spécification ONNX, vous permettant d'adapter les anciens modèles à la version actuelle ou inversement. |
Dernière mise à jour : Jeudi, le 19 septembre 2024