Yolo Vision Shenzhen
Shenzhen
Rejoindre maintenant

Entraînement personnalisé des modèlesYOLO d'Ultralytics sur Lightning AI

Abirami Vina

3 min de lecture

13 janvier 2025

Découvrez comment Lightning AI, présenté à YOLO Vision 2024, simplifie le développement évolutif de l'IA visionnaire en accélérant la formation, le déploiement et la collaboration des modèles.

Que vous soyez un développeur d'IA expérimenté ou que vous commenciez à explorer l'IA visionnaire, le fait de disposer d'un environnement fiable pour jouer et expérimenter avec des modèles de vision artificielle tels que Ultralytics YOLO11 est essentiel. Un environnement fait référence aux outils, aux ressources et à l'infrastructure nécessaires pour concevoir, tester et déployer efficacement des modèles d'IA. 

Bien que plusieurs plateformes en ligne offrent différents outils d'IA, beaucoup ne fournissent pas un environnement unifié pour l'ensemble du cycle de vie de l'IA, de la préparation des données au déploiement du modèle. C'est là qu'intervient Lightning AI, une plateforme tout-en-un pour le développement de l'IA, afin de rationaliser le processus de la préparation des données au déploiement.

La pertinence de faciliter le développement de l'IA a été mise en évidence lors de YOLO Vision 2024 (YV24), un événement hybride annuel organisé par Ultralytics qui s'est concentré sur les avancées en matière d'IA et de vision par ordinateur. Luca Antiga, directeur technique de Lightning AI, a donné une conférence intitulée" Going YOLO on Lightning Studios", au cours de laquelle il a expliqué comment entraîner les modèlesYOLO 'Ultralytics rapidement, en douceur et sans s'impliquer dans les complexités techniques à l'aide de Lightning AI.

Dans cet article, nous allons nous plonger dans les points clés de la conférence de Luca, en couvrant tout, des applications de vision par ordinateur du monde réel aux démonstrations en direct sur l'entraînement et le déploiement des modèlesYOLO d'Ultralytics avec Lightning AI. C'est parti !

Utiliser Lightning AI et Ultralytics YOLO pour simplifier le développement de l'IA

Luca a commencé son discours en partageant ses réflexions et son appréciation de l'influence des modèles YOLO dans diverses industries. Il a souligné comment les modèles YOLO peuvent être appliqués dans des secteurs tels que l'industrie manufacturière et l'agriculture. Il a déclaré : "J'apprécie l'impact que YOLO a eu sur la communauté des constructeurs - les personnes qui doivent résoudre des problèmes concrets et pratiques - cela me touche de très près".

Parallèlement à l'intérêt croissant pour la formation en IA, il a présenté Lightning AI, une plateforme conçue pour rendre le développement de modèles d'IA plus rapide, plus simple et plus accessible à tous. Elle est particulièrement utile pour soutenir les avancées itératives en IA, en aidant les développeurs à affiner et à améliorer les modèles.

Fig. 1. Luca Antiga présentant à distance Lightning Studios à YV24.

Il a également souligné que Lightning AI est similaire à PyTorch Lightning, un cadre qui simplifie le processus de formation des modèles d'IA. Cependant, la différence réside dans le fait que Lightning AI est une plateforme plus complète qui fournit un ensemble plus large d'outils et de capacités pour l'ensemble du processus de développement de l'IA, et pas seulement pour l'entraînement des modèles d'IA. 

Un composant essentiel de Lightning AI est Lightning Studios, qui offre un espace de travail intuitif pour concevoir, entraîner et déployer des modèles d'IA, rendant l'ensemble du flux de travail transparent et efficace. Vous pouvez considérer Lightning Studios comme un environnement de développement reproductible pour l'IA qui s'exécute sur le cloud. Par exemple, il offre un environnement de type Jupyter Notebook qui peut être dupliqué et partagé avec un autre développeur, ce qui contribue à améliorer la collaboration. 

Luca a ensuite expliqué les avantages de Lightning Studios : "La réplication de votre environnement n'est plus un problème. Si vous devez passer d'une machine CPU [Central Processing Unit] à une machine GPU [Graphics Processing Unit] ou lancer une formation sur un millier de machines, votre environnement sera persistant."

Configuration de Lightning Studios pour l'entraînement et le développement

Ensuite, Luca a démontré la rapidité avec laquelle vous pouvez démarrer avec Lightning Studios. En quelques clics, vous pouvez ouvrir un nouveau studio et accéder à des outils et des environnements tels que Jupyter Notebooks et VS Code, tous configurés et prêts à coder. Il a montré à quel point il était facile de passer d'une machine à l'autre. Si la tâche sur laquelle vous travaillez exige plus de puissance, vous pouvez facilement passer d'un CPU à un GPU plus puissant. Le GPU ne reste actif que lorsqu'il est utilisé ; sinon, il passe en mode veille, ce qui vous permet d'économiser vos crédits.

Luca a également mentionné les avantages de l'utilisation des modèles Studio. Ce sont des environnements de codage d'IA qui sont préfabriqués par la communauté, et vous pouvez les utiliser sans avoir à configurer quoi que ce soit. La mise en place d'un environnement pour les projets d'IA peut prendre du temps, et les modèles Studio peuvent aider à augmenter la productivité. Ces environnements sont préchargés avec tout ce qui est nécessaire pour les projets d'IA, comme les dépendances installées, les poids des modèles, les données, le code, etc.

Fig. 2. Luca expliquant ce que sont les modèles Studio.

Formation des modèlesYOLO d'Ultralytics sur Lightning Studios

Luca est ensuite passé à la démonstration en direct, en soulignant comment vous pouvez utiliser Lightning Studio pour entraîner les modèlesYOLO d 'Ultralytics . Il a ouvert un Template Studio, qui avait déjà toutes les dépendances installées, et a fait tourner une machine avec quatre GPU pour accélérer le processus d'entraînement. En ce qui concerne les données, il a expliqué que vous pouvez choisir de les stocker directement sur la machine ou de les diffuser à partir du cloud, ce qui rend le processus de formation plus rapide et plus efficace.

En quelques secondes, la machine était prête et Luca a rapidement lancé la session de formation. Au cours de la démonstration, un problème mineur a provoqué l'arrêt inattendu de la machine, mais Lightning Studios a repris de manière transparente là où elle s'était arrêtée, s'assurant qu'aucun progrès n'était perdu. Luca a souligné comment cette fiabilité favorise des flux de travail fluides, même face à des interruptions imprévues.

Poursuivant avec la démonstration, il a montré à quel point il est facile de surveiller la progression de l'entraînement à l'aide de TensorBoard, un outil de visualisation des métriques d'apprentissage automatique en temps réel. Lightning Studio simplifie encore davantage cette tâche en générant automatiquement des URL qui permettent à vous ou à vos coéquipiers du même espace de travail d'accéder aux vues TensorBoard sans aucune configuration supplémentaire. Cela rationalise la collaboration et permet à chacun de rester sur la même longueur d'onde. 

Fig. 3. Organigramme de la formation des modèlesYOLO d'Ultralytics sur Lightning Studios. Image de l'auteur.

Déployer les modèlesYOLO d'Ultralytics avec Lit Serve

Après la démonstration, Luca a recentré la discussion sur un nouveau projet, LitServe, récemment lancé par Lightning AI. LitServe simplifie le processus de transformation d'un modèle entraîné en un service scalable que d'autres peuvent utiliser, éliminant ainsi le besoin de pipelines de déploiement complexes. Il est conçu pour gérer tout, de l'empaquetage du modèle à son déploiement, avec un minimum d'effort.

Pour montrer cela en temps réel, Luca a fait une démonstration rapide à l'aide d'un modèle pré-entraîné d'analyse de l'information de l'entreprise (Ultralytics). Ultralytics YOLOv8 pré-entraîné. Il a pu créer une API simple pour traiter les demandes entrantes et renvoyer des prédictions d'images en quelques secondes. Cela signifie que n'importe qui peut envoyer une image à cette API et recevoir presque instantanément des résultats pour des tâches de vision artificielle telles que la détection d'objets. En coulisses, le modèleYOLOv8 d'Ultralytics est déployé en tant que service, gérant efficacement les requêtes, traitant les images et fournissant des prédictions avec une latence minimale.

Fig 4. Luca présentant LitServe de Lightning AI lors de YV24.

Il a effectué une inférence sur une image de pizza et Ultralytics YOLOv8 a identifié avec succès des objets tels que la pizza, une cuillère et une table à manger. Il a expliqué que si la première requête prend un peu plus de temps en raison d'un "démarrage à froid", les requêtes suivantes sont beaucoup plus rapides une fois que le système est réchauffé.

Luca a ensuite demandé : "Que se passe-t-il si je veux exposer cela au monde extérieur ?" Il a expliqué comment le plugin API Builder simplifie la transformation de votre modèle en un service en direct, prêt pour la production. Grâce à des fonctionnalités telles que les domaines personnalisés, la sécurité renforcée et l'intégration transparente, vous pouvez facilement rendre votre modèle accessible à tous.

Principaux avantages de l'utilisation de Lightning Studios

En conclusion de son exposé, Luca a évoqué l'évolutivité et la flexibilité de Lightning Studio pour le développement de l'IA. Il a mentionné comment la plateforme peut entraîner des modèles sur plusieurs machines, en passant à l'échelle jusqu'à 10 000 nœuds, avec une formation tolérante aux pannes qui reprend automatiquement après toute interruption. 

Par exemple, si une tâche d'entraînement sur un cluster GPU est interrompue en raison d'un problème matériel ou d'un redémarrage du serveur, Lightning Studios s'assure que le processus reprend exactement là où il s'est arrêté. Cela le rend idéal pour les projets d'IA à grande échelle, comme l'entraînement de modèles d'apprentissage profond sur des ensembles de données massifs tels qu'ImageNet ou COCO.

Voici d'autres avantages clés de Lightning Studios dont Luca a parlé :

  • Crédits GPU mensuels gratuits: Les utilisateurs reçoivent 15 crédits GPU gratuits chaque mois, qui se remplissent automatiquement, ce qui vous permet d'expérimenter et de développer sans coûts supplémentaires.
  • Collaboration améliorée : Les espaces d’équipe partagés et les environnements reproductibles de Lightning Studio permettent aux membres de l’équipe de travailler ensemble de manière transparente, assurant ainsi la cohérence et l’efficacité des projets.
  • Options d'instances flexibles: Il vous offre la possibilité de choisir entre des instances interruptibles et non interruptibles, ce qui permet aux utilisateurs de réduire les coûts sur les machines GPU avec des options interruptibles.
  • Intégration aux outils existants : La plateforme s'intègre aux outils de développement à distance tels que SSH (Secure Socket Shell) et VS Code, offrant une flexibilité permettant de travailler localement ou dans le cloud.

Principaux points à retenir

Le discours de Luca à YV24 a mis en évidence la façon dont l'IA, associée à des outils tels que les modèlesYOLO Ultralytics et Lightning AI, modifie la façon dont nous résolvons les problèmes du monde réel. Ils permettent aux développeurs de former et de déployer plus facilement des modèles conçus pour résoudre des problèmes spécifiques dans un grand nombre de secteurs.

Il a montré comment Lightning Studios rend l'ensemble du processus de développement plus rapide et plus accessible, permettant aux développeurs de créer facilement des solutions puissantes. Au cœur de plateformes de pointe comme Lightning AI, les modèles de vision par ordinateur transforment la manière dont les solutions d'IA relèvent les défis. En particulier, avec le dernier modèle Ultralytics YOLO11 , les développeurs peuvent créer des solutions qui ont un impact significatif.

Rejoignez notre communauté pour rester informé sur l'IA et ses utilisations pratiques. Consultez notre dépôt GitHub pour explorer les innovations dans des secteurs tels que l'IA dans la conduite autonome et la vision par ordinateur dans le secteur de la santé.

Construisons ensemble l'avenir
de l'IA !

Commencez votre parcours avec l'avenir de l'apprentissage automatique

Démarrer gratuitement