Link to this sectionGuide sur l'exportation du modèle YOLO26 vers TFLite pour le déploiement#
Le déploiement de modèles de vision par ordinateur sur des appareils de périphérie ou embarqués nécessite un format capable de garantir des performances fluides.
Le format d'exportation TensorFlow Lite ou TFLite te permet d'optimiser tes modèles Ultralytics YOLO26 pour des tâches comme la détection d'objets et la classification d'images dans des applications sur appareils de périphérie. Dans ce guide, nous parcourrons les étapes de conversion de tes modèles vers le format TFLite, facilitant ainsi leur exécution performante sur divers appareils de périphérie.
Link to this sectionPourquoi exporter vers TFLite ?#
Introduit par Google en mai 2017 dans le cadre de son framework TensorFlow, TensorFlow Lite, ou TFLite en abrégé, est un framework de deep learning open-source conçu pour l'inférence sur appareil, aussi connu sous le nom d'edge computing. Il fournit aux développeurs les outils nécessaires pour exécuter leurs modèles entraînés sur des appareils mobiles, embarqués et IoT, ainsi que sur des ordinateurs traditionnels.
TensorFlow Lite est compatible avec un large éventail de plateformes, y compris Linux embarqué, Android, iOS et les microcontrôleurs (MCU). Exporter ton modèle vers TFLite rend tes applications plus rapides, plus fiables et capables de fonctionner hors ligne.
Link to this sectionFonctionnalités clés des modèles TFLite#
Les modèles TFLite offrent un large éventail de fonctionnalités clés qui permettent l'apprentissage automatique sur appareil en aidant les développeurs à exécuter leurs modèles sur des appareils mobiles, embarqués et de périphérie :
-
Optimisation sur appareil : TFLite optimise pour le ML sur appareil, réduisant la latence en traitant les données localement, améliorant la confidentialité en ne transmettant pas de données personnelles, et minimisant la taille du modèle pour économiser de l'espace.
-
Prise en charge multiplateforme : TFLite offre une compatibilité étendue avec les plateformes, prenant en charge Android, iOS, Linux embarqué et les microcontrôleurs.
-
Prise en charge de divers langages : TFLite est compatible avec divers langages de programmation, dont Java, Swift, Objective-C, C++ et Python.
-
Haute performance : Atteint des performances supérieures grâce à l'accélération matérielle et à l'optimisation des modèles.
Link to this sectionOptions de déploiement dans TFLite#
Avant d'examiner le code pour exporter les modèles YOLO26 au format TFLite, comprenons comment les modèles TFLite sont normalement utilisés.
TFLite propose diverses options de déploiement sur appareil pour les modèles d'apprentissage automatique, notamment :
- Déploiement avec Android et iOS : Les applications Android et iOS avec TFLite peuvent analyser les flux de caméras et les capteurs en périphérie pour détecter et identifier des objets. TFLite propose également des bibliothèques iOS natives écrites en Swift et Objective-C. Le schéma d'architecture ci-dessous montre le processus de déploiement d'un modèle entraîné sur des plateformes Android et iOS à l'aide de TensorFlow Lite.
-
Mise en œuvre avec Linux embarqué : Si l'exécution des inférences sur un Raspberry Pi en utilisant le Guide Ultralytics ne répond pas aux exigences de vitesse de ton cas d'utilisation, tu peux utiliser un modèle TFLite exporté pour accélérer les temps d'inférence. De plus, il est possible d'améliorer encore les performances en utilisant un appareil Coral Edge TPU.
-
Déploiement avec des microcontrôleurs : Les modèles TFLite peuvent également être déployés sur des microcontrôleurs et d'autres appareils avec seulement quelques kilo-octets de mémoire. Le runtime de base tient dans 16 Ko sur un Arm Cortex M3 et peut exécuter de nombreux modèles basiques. Il ne nécessite pas de support de système d'exploitation, de bibliothèques C ou C++ standard, ni d'allocation dynamique de mémoire.
Link to this sectionExporter vers TFLite : Conversion de ton modèle YOLO26#
Tu peux améliorer l'efficacité de l'exécution du modèle sur appareil et optimiser les performances en convertissant tes modèles au format TFLite.
Link to this sectionInstallation#
Pour installer les paquets requis, exécute :
# Install the required package for YOLO26
pip install ultralyticsPour 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 des solutions et des conseils.
Link to this sectionUtilisation#
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 workflow de déploiement préféré. Tu peux consulter la liste complète des formats d'exportation pris en charge et des options de configuration pour choisir la meilleure configuration pour ton application.
Le format TFLite prend en charge les modes Export, Predict et Validate. Exporte ton modèle, puis charge le modèle exporté pour exécuter l'inférence ou valider sa précision.
from ultralytics import YOLO
# Load a YOLO26 model
model = YOLO("yolo26n.pt")
# Export the model to TFLite format
model.export(format="tflite") # creates 'yolo26n_float32.tflite'from ultralytics import YOLO
# Load the exported TFLite model
model = YOLO("yolo26n_float32.tflite")
# Run inference
results = model("https://ultralytics.com/images/bus.jpg")from ultralytics import YOLO
# Load the exported TFLite model
model = YOLO("yolo26n_float32.tflite")
# Validate accuracy on the COCO8 dataset
metrics = model.val(data="coco8.yaml")Link to this sectionArguments d'exportation#
| Argument | Type | Défaut | Description |
|---|---|---|---|
format | str | 'tflite' | Format cible pour le modèle exporté, définissant la compatibilité avec divers environnements de déploiement. |
imgsz | int ou tuple | 640 | Taille d'image souhaitée pour l'entrée du modèle. Peut être un entier pour des images carrées ou un tuple (height, width) pour des dimensions spécifiques. |
half | bool | False | Active la quantification FP16 (demi-précision), réduisant la taille du modèle et accélérant potentiellement l'inférence sur le matériel pris en charge. |
int8 | bool | False | Active la quantification INT8, compressant davantage le modèle et accélérant l'inférence avec une perte de précision minimale, principalement pour les appareils en périphérie. |
nms | bool | False | Ajoute le NMS (Non-Maximum Suppression), essentiel pour un post-traitement de détection précis et efficace. |
batch | int | 1 | Spécifie la taille de l'inférence par lot du modèle exporté ou le nombre maximal d'images que le modèle exporté traitera simultanément en mode predict. |
data | str | 'coco8.yaml' | Chemin vers le fichier de configuration du dataset (par défaut : coco8.yaml), essentiel pour la quantification. |
fraction | float | 1.0 | Spécifie la fraction du jeu de données à utiliser pour le calibrage de la quantification INT8. Permet le calibrage sur un sous-ensemble du jeu de données complet, utile pour des expériences ou lorsque les ressources sont limitées. S'il n'est pas spécifié avec INT8 activé, le jeu de données complet sera utilisé. |
device | str | None | Spécifie l'appareil 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.
Link to this sectionDéploiement des modèles YOLO26 TFLite exportés#
Après avoir exporté avec succès tes modèles Ultralytics YOLO26 vers le format TFLite, tu peux maintenant les déployer. La première étape principale et recommandée pour exécuter un modèle TFLite est d'utiliser la méthode YOLO("model.tflite"), comme indiqué dans l'extrait de code d'utilisation précédent. Cependant, pour des instructions approfondies sur le déploiement de tes modèles TFLite dans divers autres contextes, jette un œil aux ressources suivantes :
-
Android: A quick start guide for integrating TensorFlow Lite into Android applications, providing easy-to-follow steps for setting up and running machine learning models.
-
iOS : Consulte ce guide détaillé pour les développeurs sur l'intégration et le déploiement de modèles TensorFlow Lite dans des applications iOS, offrant des instructions étape par étape et des ressources.
-
Exemples de bout en bout : Cette page donne un aperçu de divers exemples TensorFlow Lite, présentant des applications pratiques et des tutoriels conçus pour aider les développeurs à implémenter TensorFlow Lite dans leurs projets d'apprentissage automatique sur appareils mobiles et de périphérie.
Link to this sectionRésumé#
Dans ce guide, nous nous sommes concentrés sur la manière d'exporter vers le format TFLite. En convertissant tes modèles Ultralytics YOLO26 au format de modèle TFLite, tu peux améliorer l'efficacité et la vitesse des modèles YOLO26, les rendant plus performants et adaptés aux environnements d'edge computing.
Pour plus de détails sur l'utilisation, visite la documentation officielle de TFLite.
De plus, si tu es curieux à propos d'autres intégrations Ultralytics YOLO26, consulte notre page de guide d'intégration. Tu y trouveras beaucoup d'informations et d'aperçus utiles.
Link to this sectionFAQ#
Link to this sectionComment exporter un modèle YOLO26 au format TFLite ?#
Pour exporter un modèle YOLO26 au format TFLite, tu peux utiliser la bibliothèque Ultralytics. Tout d'abord, installe le package requis en utilisant :
pip install ultralyticsEnsuite, utilise l'extrait de code suivant pour exporter ton modèle :
from ultralytics import YOLO
# Load a YOLO26 model
model = YOLO("yolo26n.pt")
# Export the model to TFLite format
model.export(format="tflite") # creates 'yolo26n_float32.tflite'Pour les utilisateurs de CLI, tu peux y parvenir avec :
yolo export model=yolo26n.pt format=tflite # creates 'yolo26n_float32.tflite'Pour plus de détails, visite le guide d'exportation Ultralytics.
Link to this sectionQuels sont les avantages d'utiliser TensorFlow Lite pour le déploiement du modèle YOLO26 ?#
TensorFlow Lite (TFLite) est un framework de deep learning open-source conçu pour l'inférence sur appareil, le rendant idéal pour déployer des modèles YOLO26 sur des appareils mobiles, embarqués et IoT. Les avantages clés incluent :
- Optimisation sur appareil : Minimise la latence et améliore la confidentialité en traitant les données localement.
- Compatibilité de plateforme : Prend en charge Android, iOS, Linux embarqué et MCU.
- Performance : Utilise l'accélération matérielle pour optimiser la vitesse et l'efficacité du modèle.
Pour en savoir plus, consulte le guide TFLite.
Link to this sectionEst-il possible d'exécuter des modèles YOLO26 TFLite sur Raspberry Pi ?#
Oui, tu peux exécuter des modèles YOLO26 TFLite sur Raspberry Pi pour améliorer les vitesses d'inférence. D'abord, exporte ton modèle au format TFLite comme expliqué ci-dessus. Ensuite, utilise un outil comme TensorFlow Lite Interpreter pour exécuter le modèle sur ton Raspberry Pi.
Pour d'autres optimisations, tu pourrais envisager d'utiliser Coral Edge TPU. Pour des étapes détaillées, reporte-toi à notre guide de déploiement Raspberry Pi et au guide d'intégration Edge TPU.
Link to this sectionPuis-je utiliser des modèles TFLite sur des microcontrôleurs pour des prédictions YOLO26 ?#
Oui, TFLite prend en charge le déploiement sur des microcontrôleurs avec des ressources limitées. Le runtime de base de TFLite ne nécessite que 16 Ko de mémoire sur un Arm Cortex M3 et peut exécuter des modèles YOLO26 basiques. Cela le rend adapté au déploiement sur des appareils avec une puissance de calcul et une mémoire minimales.
Pour commencer, visite le guide TFLite Micro pour microcontrôleurs.
Link to this sectionQuelles plateformes sont compatibles avec les modèles YOLO26 exportés en TFLite ?#
TensorFlow Lite offre une compatibilité étendue avec les plateformes, te permettant de déployer des modèles YOLO26 sur une large gamme d'appareils, notamment :
- Android et iOS : Support natif via les bibliothèques Android et iOS TFLite.
- Linux embarqué : Idéal pour les ordinateurs monocarte tels que Raspberry Pi.
- Microcontrôleurs : Adapté aux MCU avec des ressources limitées.
Pour plus d'informations sur les options de déploiement, consulte notre guide de déploiement détaillé.
Link to this sectionComment résoudre les problèmes courants lors de l'exportation du modèle YOLO26 vers TFLite ?#
Si tu rencontres des erreurs lors de l'exportation de modèles YOLO26 vers TFLite, les solutions courantes incluent :
- Vérifier la compatibilité des packages : Assure-toi d'utiliser des versions compatibles d'Ultralytics et de TensorFlow. Reporte-toi à notre guide d'installation.
- Support du modèle : Vérifie que le modèle YOLO26 spécifique prend en charge l'exportation TFLite en consultant la page de documentation d'exportation d'Ultralytics.
- Problèmes de quantification : Lorsque tu utilises la quantification INT8, assure-toi que le chemin de ton jeu de données est correctement spécifié dans le paramètre
data.
Pour des conseils de dépannage supplémentaires, visite notre guide des problèmes courants.