Contrôle vert
Lien copié dans le presse-papiers

Exploration de Ultralytics YOLOv8 's ML Experiment Tracking Integrations

Apprends à mieux connaître les différentes options pour suivre et contrôler tes expériences de formation au modèle YOLOv8 . Compare les outils et trouve celui qui correspond le mieux à tes besoins.

La collecte de données, leur annotation et l'entraînement de modèles tels que le modèleUltralytics YOLOv8 sont au cœur de tout projet de vision par ordinateur. Souvent, tu devras entraîner ton modèle personnalisé plusieurs fois avec différents paramètres pour créer le modèle le plus optimal. L'utilisation d'outils pour suivre tes expériences de formation peut rendre la gestion de ton projet de vision par ordinateur un peu plus facile. Le suivi des expériences consiste à enregistrer les détails de chaque formation, comme les paramètres utilisés, les résultats obtenus et les modifications apportées en cours de route. 

Fig 1. Une image montrant comment le suivi des expériences s'intègre dans un projet de vision par ordinateur. 

Garder une trace de ces détails t'aide à reproduire tes résultats, à comprendre ce qui fonctionne et ce qui ne fonctionne pas, et à affiner tes modèles plus efficacement. Pour les organisations, cela permet de maintenir la cohérence entre les équipes, de favoriser la collaboration et de fournir une piste de vérification claire. Pour les individus, il s'agit de maintenir une documentation claire et organisée de ton travail qui te permet d'affiner ton approche et d'obtenir de meilleurs résultats au fil du temps. 

Dans cet article, nous allons te présenter les différentes intégrations de formation disponibles pour gérer et surveiller tes YOLOv8 expériences. Que tu travailles seul ou au sein d'une équipe plus importante, la compréhension et l'utilisation des bons outils de suivi peuvent faire une réelle différence dans la réussite de tes projetsYOLOv8 .

Suivi des expériences d'apprentissage automatique avec MLflow

MLflow est une plateforme open-source développée par Databricks qui facilite la gestion de l'ensemble du cycle de vie de l'apprentissage automatique. MLflow Tracking est un composant essentiel de MLflow qui fournit une API et une interface utilisateur qui aide les scientifiques des données et les ingénieurs à consigner et à visualiser leurs expériences d'apprentissage automatique. Il prend en charge plusieurs langages et interfaces, notamment Python, REST, Java et R API. 

Le suivi MLflow s'intègre parfaitement à YOLOv8, et tu peux enregistrer des mesures importantes comme la précision, le rappel et la perte directement à partir de tes modèles. La configuration de MLflow avec YOLOv8 est simple, et il existe des options flexibles : tu peux utiliser la configuration locale par défaut, te connecter à différents magasins de données ou démarrer un serveur de suivi MLflow à distance pour que tout soit organisé.

Fig 2. Configurations courantes de l'environnement de suivi MLflow. Source de l'image : MLflow tracking.

Voici quelques apports pour t'aider à décider si MLflow est le bon outil pour ton projet :

  • Évolutivité : MLflow évolue bien en fonction de tes besoins, que tu travailles sur une seule machine ou que tu déploies sur de grands clusters. Si ton projet implique une mise à l'échelle du développement à la production, MLflow peut prendre en charge cette croissance.
  • Complexité du projet: MLflow est idéal pour les projets complexes qui nécessitent un suivi approfondi, une gestion des modèles et des capacités de déploiement. Si ton projet nécessite ces fonctionnalités à grande échelle, MLflow peut rationaliser tes flux de travail.
  • Installation et maintenance : Bien que puissant, MLflow s'accompagne d'une courbe d'apprentissage et de frais d'installation. 

Utilisation de Weights & Biases (W&B) pour le suivi de modèles de vision par ordinateur

Weights & Biases est une plateforme MLOps pour le suivi, la visualisation et la gestion des expériences d'apprentissage automatique. En utilisant W&B avec YOLOv8, tu peux surveiller les performances de tes modèles au fur et à mesure que tu les entraînes et que tu les affines. Le tableau de bord interactif de W&B offre une vue claire et en temps réel de ces mesures et permet de repérer plus facilement les tendances, de comparer les variantes de modèles et de résoudre les problèmes pendant le processus de formation.

W&B enregistre automatiquement les mesures de formation et les points de contrôle du modèle, et tu peux même l'utiliser pour affiner les hyperparamètres comme le taux d'apprentissage et la taille du lot. La plateforme prend en charge un large éventail d'options de configuration, du suivi des exécutions sur ta machine locale à la gestion de projets à grande échelle avec un stockage dans le nuage.

Fig 3. Un exemple de tableau de bord de suivi d'expériences Weights & Biases'. Source de l'image : Weights & Biases track experiments.

Voici quelques éléments pour t'aider à décider si Weights & Biases est le bon outil pour ton projet :

  • Visualisation et suivi améliorés: W&B propose un tableau de bord intuitif pour visualiser les mesures de formation et modéliser les performances en temps réel. 
  • Modèle de tarification: La tarification est basée sur les heures suivies, ce qui n'est peut-être pas idéal pour les utilisateurs dont le budget est limité ou pour les projets qui impliquent de longs temps de formation.

Suivi des expériences MLOps avec ClearML

ClearML est une plateforme MLOps open-source conçue pour automatiser, surveiller et orchestrer les flux de travail d'apprentissage automatique. Elle prend en charge les frameworks d'apprentissage automatique les plus populaires comme PyTorch, TensorFlow, et Keras et peut s'intégrer facilement à tes processus existants. ClearML prend également en charge l'informatique distribuée sur des machines locales ou dans le cloud et peut surveiller l'utilisation de CPU et GPU .

YOLOv8L'intégration de ClearML fournit des outils pour le suivi des expériences, la gestion des modèles et la surveillance des ressources. L'interface Web intuitive de la plateforme te permet de visualiser les données, de comparer les expériences et de suivre en temps réel les paramètres critiques tels que la perte, la précision et les scores de validation. L'intégration prend également en charge des fonctions avancées telles que l'exécution à distance, le réglage des hyperparamètres et le point de contrôle du modèle.

Fig 4. Exemple de visualisation du suivi des expériences sur ClearML. Source de l'image : Clear ML Tracking Experiments and Visualizing Results.

Voici quelques éléments pour t'aider à décider si ClearML est le bon outil pour ton projet :

  • Besoin d'un suivi avancé des expériences : ClearML offre un suivi robuste des expériences qui comprend une intégration automatique avec Git. 
  • Déploiement flexible: ClearML peut être utilisé sur site, dans le cloud ou sur des clusters Kubernetes, ce qui le rend adaptable à différentes configurations.

Suivre les expériences de formation à l'aide de Comet ML

Comet ML est une plateforme conviviale qui aide à gérer et à suivre les expériences d'apprentissage automatique. YOLOv8L'intégration d'IMSMA avec Comet ML te permet d'enregistrer tes expériences et d'afficher tes résultats au fil du temps. L'intégration permet de repérer plus facilement les tendances et de comparer différentes expériences. 

Comet ML peut être utilisé dans le nuage, sur un nuage privé virtuel (VPC) ou même sur place, ce qui le rend adaptable à différentes configurations et à différents besoins. Cet outil est conçu pour le travail en équipe. Tu peux partager des projets, étiqueter des coéquipiers et laisser des commentaires pour que tout le monde puisse rester sur la même longueur d'onde et reproduire les expériences avec précision.

Voici quelques données d'entrée pour t'aider à décider si Comet ML est le bon outil pour ton projet :

  • Prend en charge plusieurs cadres et langages : Comet ML fonctionne avec PythonIl s'agit donc d'une option polyvalente, quels que soient les outils ou les langages utilisés dans le cadre de ton projet.
  • Tableaux de bord et rapports personnalisables: Comet L'interface de ML est hautement personnalisable, tu peux donc créer les rapports et les tableaux de bord qui ont le plus de sens pour ton projet. 
  • Coût: Comet ML est une plateforme commerciale, et certaines de ses fonctions avancées nécessitent un abonnement payant.

TensorBoard peut aider à la visualisation

TensorBoard est une puissante boîte à outils de visualisation spécialement conçue pour les expériences TensorFlow , mais c'est aussi un excellent outil pour suivre et visualiser les métriques d'un large éventail de projets d'apprentissage automatique. Connu pour sa simplicité et sa rapidité, TensorBoard permet aux utilisateurs de suivre facilement les métriques clés et de visualiser les graphiques de modèles, les embeddings et d'autres types de données.

L'un des principaux avantages de l'utilisation de TensorBoard avec YOLOv8 est qu'il est commodément préinstallé, ce qui élimine le besoin d'une configuration supplémentaire. Un autre avantage est la capacité de TensorBoard à fonctionner entièrement sur site. Ceci est particulièrement important pour les projets ayant des exigences strictes en matière de confidentialité des données ou pour ceux qui se trouvent dans des environnements où les téléchargements dans le nuage ne sont pas une option.

Fig 5. Suivi de la formation du modèle YOLOv8 à l'aide de TensorBoard.

Voici quelques apports pour t'aider à décider si TensorBoard est le bon outil pour ton projet :

  • Explicabilité avec l'outil What-If (WIT): TensorBoard comprend l'outil What-If, qui offre une interface facile à utiliser pour explorer et comprendre les modèles ML. Il est précieux pour ceux qui cherchent à mieux comprendre les modèles de boîte noire et à améliorer l'explicabilité.
  • Suivi simple des expériences: TensorBoard est idéal pour les besoins de suivi de base avec une comparaison d'expériences limitée et manque de fonctions robustes de collaboration d'équipe, de contrôle de version et de gestion de la confidentialité.

Utilisation de DVCLive (Data Version Control Live) pour suivre les expériences de ML

YOLOv8L' intégration de DVCLive à DVCLive offre un moyen simplifié de suivre et de gérer les expériences en donnant une version à tes ensembles de données, à tes modèles et à ton code sans avoir à stocker de gros fichiers dans Git. Il utilise des commandes similaires à celles de Git et stocke les mesures suivies dans des fichiers texte pour faciliter le contrôle des versions. DVCLive enregistre les mesures clés, visualise les résultats et gère les expériences proprement sans encombrer ton référentiel. Il prend en charge un large éventail de fournisseurs de stockage et peut fonctionner localement ou dans le cloud. DVCLive est parfait pour les équipes qui cherchent à rationaliser le suivi des expériences sans infrastructure supplémentaire ou dépendance au cloud.

Gérer les modèles et les flux de travail de Ultralytics à l'aide de Ultralytics HUB

Ultralytics HUB est une plateforme interne tout-en-un conçue pour simplifier la formation, le déploiement et la gestion desmodèles Ultralytics YOLO tels que YOLOv5 et YOLOv8. Contrairement aux intégrations externes, Ultralytics HUB offre une expérience transparente et native créée spécifiquement pour les utilisateurs de YOLO . Il simplifie l'ensemble du processus, te permettant de télécharger facilement des ensembles de données, de choisir des modèles pré-entraînés et de commencer l'entraînement en quelques clics en utilisant les ressources du cloud - le tout dans l'interface conviviale du HUB. UltralyticsHUB prend également en charge le suivi des expériences, ce qui facilite le contrôle de la progression de la formation, la comparaison des résultats et l'ajustement des modèles.

Fig 7. Suivi de l'entraînement du modèle YOLOv8 à l'aide de Ultralytics HUB.

Principaux enseignements

Choisir le bon outil pour suivre tes expériences d'apprentissage automatique peut faire une grande différence. Tous les outils dont nous avons parlé peuvent t'aider à suivre les expériences d'apprentissage YOLOv8 , mais il est important de peser le pour et le contre de chacun d'entre eux pour trouver celui qui convient le mieux à ton projet. Le bon outil te permettra de rester organisé et d'améliorer les performances de ton modèle YOLOv8 ! 

Les intégrations peuvent simplifier l'utilisation de YOLOv8 dans tes projets innovants et accélérer tes progrès. Pour explorer d'autres intégrations passionnantes de YOLOv8 , consulte notre documentation.

Apprends à en savoir plus sur l'IA en explorant notre dépôt GitHub et en rejoignant notre communauté. Consulte nos pages sur les solutions pour obtenir un aperçu détaillé de l'IA dans les secteurs de la fabrication et de la santé. 🚀

Logo FacebookLogo de TwitterLogo LinkedInSymbole du lien de copie

Lire la suite dans cette catégorie

Construisons ensemble le futur
de l'IA !

Commence ton voyage avec le futur de l'apprentissage automatique.