Comment exporter vers TF GraphDef depuis YOLO26 pour le déploiement

Lorsque tu déploies des modèles de computer vision de pointe, comme YOLO26, dans différents environnements, tu peux rencontrer des problèmes de compatibilité. Le format TensorFlow GraphDef de Google, ou TF GraphDef, offre une solution en proposant une représentation sérialisée et indépendante de la plateforme pour ton modèle. En utilisant le format de modèle TF GraphDef, tu peux déployer ton modèle YOLO26 dans des environnements où l'écosystème complet de TensorFlow n'est pas forcément disponible, comme sur des appareils mobiles ou du matériel spécialisé.

Dans ce guide, nous allons te montrer étape par étape comment exporter tes modèles Ultralytics YOLO26 vers le format de modèle TF GraphDef. En convertissant ton modèle, tu peux simplifier le déploiement et utiliser les capacités de computer vision de YOLO26 dans un éventail plus large d'applications et de plateformes.

TensorFlow GraphDef model serialization format

Pourquoi devrais-tu exporter vers TF GraphDef ?

TF GraphDef est un composant puissant de l'écosystème TensorFlow développé par Google. Il peut être utilisé pour optimiser et déployer des modèles comme YOLO26. L'exportation vers TF GraphDef te permet de faire passer tes modèles de la recherche à des applications concrètes. Cela permet aux modèles de s'exécuter dans des environnements sans le framework TensorFlow complet.

Le format GraphDef représente le modèle sous forme de graphe de calcul sérialisé. Cela permet diverses techniques d'optimisation comme le pliage de constantes, la quantification et les transformations de graphes. Ces optimisations garantissent une exécution efficace, une consommation mémoire réduite et des vitesses d'inférence plus rapides.

Les modèles GraphDef peuvent utiliser des accélérateurs matériels tels que des GPU, des TPU et des puces IA, débloquant des gains de performance significatifs pour le pipeline d'inférence de YOLO26. Le format TF GraphDef crée un package autonome contenant le modèle et ses dépendances, simplifiant ainsi le déploiement et l'intégration dans divers systèmes.

Fonctionnalités clés des modèles TF GraphDef

TF GraphDef offre des fonctionnalités distinctes pour rationaliser le model deployment et l'optimisation.

Voici un aperçu de ses caractéristiques principales :

  • Sérialisation de modèle : TF GraphDef offre un moyen de sérialiser et de stocker des modèles TensorFlow dans un format indépendant de la plateforme. Cette représentation sérialisée te permet de charger et d'exécuter tes modèles sans la base de code Python originale, rendant le déploiement plus simple.

  • Optimisation de graphe : TF GraphDef permet l'optimisation des graphes de calcul. Ces optimisations peuvent booster les performances en rationalisant le flux d'exécution, en réduisant les redondances et en adaptant les opérations au matériel spécifique.

  • Flexibilité de déploiement : Les modèles exportés au format GraphDef peuvent être utilisés dans divers environnements, y compris des appareils aux ressources limitées, des navigateurs web et des systèmes dotés de matériel spécialisé. Cela ouvre des possibilités pour un déploiement plus large de tes modèles TensorFlow.

  • Orientation production : GraphDef est conçu pour le déploiement en production. Il prend en charge l'exécution efficace, des fonctionnalités de sérialisation et des optimisations alignées sur des cas d'utilisation réels.

Options de déploiement avec TF GraphDef

Avant de nous plonger dans le processus d'exportation des modèles YOLO26 vers TF GraphDef, examinons quelques situations de déploiement typiques où ce format est utilisé.

Voici comment tu peux déployer efficacement avec TF GraphDef sur diverses plateformes.

  • TensorFlow Serving : Ce framework est conçu pour déployer des modèles TensorFlow dans des environnements de production. TensorFlow Serving propose une gestion de modèles, un versionnage et l'infrastructure nécessaire pour une diffusion efficace des modèles à grande échelle. C'est un moyen fluide d'intégrer tes modèles basés sur GraphDef dans des services web ou des API de production.

  • Appareils mobiles et embarqués : Avec des outils comme TensorFlow Lite, tu peux convertir des modèles TF GraphDef en formats optimisés pour les smartphones, tablettes et divers appareils embarqués. Tes modèles peuvent alors être utilisés pour l'inférence sur l'appareil, où l'exécution se fait localement, offrant souvent des gains de performance et des capacités hors ligne.

  • Navigateurs Web : TensorFlow.js permet le déploiement de modèles TF GraphDef directement dans les navigateurs web. Cela ouvre la voie à des applications de détection d'objets en temps réel s'exécutant côté client, en utilisant les capacités de YOLO26 via JavaScript.

  • Matériel spécialisé : La nature agnostique vis-à-vis de la plateforme de TF GraphDef lui permet de cibler du matériel personnalisé, tel que des accélérateurs et des TPU (Tensor Processing Units). Ces appareils peuvent fournir des avantages en termes de performance pour des modèles gourmands en calcul.

Exporter des modèles YOLO26 vers TF GraphDef

Tu peux convertir ton modèle de détection d'objets YOLO26 vers le format TF GraphDef, compatible avec divers systèmes, afin d'améliorer ses performances sur différentes plateformes.

Installation

Pour installer le package requis, exécute :

Installation
# Install the required package for YOLO26
pip install ultralytics

Pour des instructions détaillées et les meilleures pratiques liées au processus d'installation, consulte notre guide d'installation Ultralytics. Lors de l'installation des packages requis pour YOLO26, si tu rencontres des difficultés, consulte notre guide des problèmes courants pour obtenir des solutions et des conseils.

Utilisation

Tous les modèles Ultralytics YOLO26 sont conçus pour prendre en charge l'exportation dès leur installation, ce qui facilite leur intégration dans ton flux de travail de déploiement préféré. Tu peux consulter la liste complète des formats d'exportation pris en charge et les options de configuration pour choisir la meilleure configuration pour ton application.

Utilisation
from ultralytics import YOLO

# Load the YOLO26 model
model = YOLO("yolo26n.pt")

# Export the model to TF GraphDef format
model.export(format="pb")  # creates 'yolo26n.pb'

# Load the exported TF GraphDef model
tf_graphdef_model = YOLO("yolo26n.pb")

# Run inference
results = tf_graphdef_model("https://ultralytics.com/images/bus.jpg")

Arguments d'exportation

ArgumentTypePar défautDescription
formatstr'pb'Format cible pour le modèle exporté, définissant la compatibilité avec divers environnements de déploiement.
imgszint ou tuple640Desired image size for the model input. Can be an integer for square images or a tuple (height, width) for specific dimensions.
batchint1Spécifie la taille d'inférence par lots du modèle exporté ou le nombre maximal d'images que le modèle exporté traitera simultanément en mode predict.
devicestrNoneSpécifie le périphérique pour l'exportation : CPU (device=cpu), MPS pour Apple silicon (device=mps).

Pour plus de détails sur le processus d'exportation, visite la page de documentation Ultralytics sur l'exportation.

Déployer des modèles YOLO26 TF GraphDef exportés

Une fois que tu as exporté ton modèle YOLO26 vers le format TF GraphDef, l'étape suivante est le déploiement. La première étape principale et recommandée pour exécuter un modèle TF GraphDef est d'utiliser la méthode YOLO("model.pb"), comme indiqué précédemment dans l'extrait de code d'utilisation.

Cependant, pour plus d'informations sur le déploiement de tes modèles TF GraphDef, jette un œil aux ressources suivantes :

  • TensorFlow Serving : Un guide sur TensorFlow Serving qui enseigne comment déployer et servir des modèles de machine learning efficacement dans des environnements de production.

  • TensorFlow Lite : Cette page décrit comment convertir des modèles de machine learning dans un format optimisé pour l'inférence sur l'appareil avec TensorFlow Lite.

  • TensorFlow.js : Un guide sur la conversion de modèles qui enseigne comment convertir des modèles TensorFlow ou Keras au format TensorFlow.js pour une utilisation dans des applications web.

Résumé

Dans ce guide, nous avons exploré comment exporter des modèles Ultralytics YOLO26 vers le format TF GraphDef. En faisant cela, tu peux déployer de manière flexible tes modèles YOLO26 optimisés dans différents environnements.

Pour plus de détails sur l'utilisation, visite la documentation officielle de TF GraphDef.

Pour plus d'informations sur l'intégration d'Ultralytics YOLO26 avec d'autres plateformes et frameworks, consulte notre page du guide d'intégration.

FAQ

Comment exporter un modèle YOLO26 vers le format TF GraphDef ?

Les modèles Ultralytics YOLO26 peuvent être exportés vers le format TensorFlow GraphDef (TF GraphDef) de manière transparente. Ce format fournit une représentation sérialisée et indépendante de la plateforme du modèle, idéale pour le déploiement dans des environnements variés comme le mobile et le web. Pour exporter un modèle YOLO26 vers TF GraphDef, suis ces étapes :

Utilisation
from ultralytics import YOLO

# Load the YOLO26 model
model = YOLO("yolo26n.pt")

# Export the model to TF GraphDef format
model.export(format="pb")  # creates 'yolo26n.pb'

# Load the exported TF GraphDef model
tf_graphdef_model = YOLO("yolo26n.pb")

# Run inference
results = tf_graphdef_model("https://ultralytics.com/images/bus.jpg")

Pour plus d'informations sur les différentes options d'exportation, visite la documentation Ultralytics sur l'exportation de modèles.

Quels sont les avantages de l'utilisation de TF GraphDef pour le déploiement de modèles YOLO26 ?

Exporter des modèles YOLO26 vers le format TF GraphDef offre de multiples avantages, notamment :

  1. Indépendance de la plateforme : TF GraphDef fournit un format indépendant de la plateforme, permettant aux modèles d'être déployés dans divers environnements, y compris les navigateurs mobiles et web.
  2. Optimisations : Le format permet plusieurs optimisations, telles que le pliage de constantes, la quantification et les transformations de graphes, qui améliorent l'efficacité de l'exécution et réduisent la consommation de mémoire.
  3. Accélération matérielle : Les modèles au format TF GraphDef peuvent tirer parti d'accélérateurs matériels comme des GPU, des TPU et des puces IA pour obtenir des gains de performance.

En savoir plus sur les avantages dans la section TF GraphDef de notre documentation.

Pourquoi devrais-je utiliser Ultralytics YOLO26 plutôt que d'autres modèles d'object detection ?

Ultralytics YOLO26 offre de nombreux avantages par rapport à d'autres modèles comme YOLOv5 et YOLOv7. Voici quelques avantages clés :

  1. Performance de pointe : YOLO26 offre une vitesse et une précision exceptionnelles pour la détection d'objets, la segmentation d'instance, la segmentation sémantique et la classification en temps réel.
  2. Facilité d'utilisation : Il dispose d'une API conviviale pour l'entraînement, la validation, la prédiction et l'exportation de modèles, le rendant accessible aussi bien aux débutants qu'aux experts.
  3. Compatibilité étendue : Prend en charge de multiples formats d'exportation dont ONNX, TensorRT, CoreML et TensorFlow, pour des options de déploiement polyvalentes.

Explore plus de détails dans notre introduction à YOLO26.

Comment puis-je déployer un modèle YOLO26 sur du matériel spécialisé en utilisant TF GraphDef ?

Une fois qu'un modèle YOLO26 est exporté au format TF GraphDef, tu peux le déployer sur diverses plateformes matérielles spécialisées. Les scénarios de déploiement typiques incluent :

  • TensorFlow Serving : Utilise TensorFlow Serving pour un déploiement de modèle évolutif dans des environnements de production. Il prend en charge la gestion des modèles et une diffusion efficace.
  • Appareils mobiles : Convertis les modèles TF GraphDef en TensorFlow Lite, optimisés pour les appareils mobiles et embarqués, permettant une inférence sur l'appareil.
  • Navigateurs Web : Déploie des modèles en utilisant TensorFlow.js pour une inférence côté client dans des applications web.
  • Accélérateurs IA : Tire parti des TPU et des puces IA personnalisées pour une inférence accélérée.

Consulte la section options de déploiement pour des informations détaillées.

Où puis-je trouver des solutions aux problèmes courants lors de l'exportation de modèles YOLO26 ?

Pour le dépannage des problèmes courants avec l'exportation de modèles YOLO26, Ultralytics propose des guides et des ressources complets. Si tu rencontres des problèmes lors de l'installation ou de l'exportation du modèle, réfère-toi à :

Ces ressources devraient t'aider à résoudre la plupart des problèmes liés à l'exportation et au déploiement de modèles YOLO26.

Commentaires