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

Segmentation d'images avec Ultralytics YOLO11 sur Google Colab

Découvre comment utiliser efficacement Ultralytics YOLO11 pour la segmentation d'images, en tirant parti d'un ensemble de données de pièces de voitures sur Google Colab pour une formation et des tests transparents.

Ultralytics YOLO comme le dernier Ultralytics YOLO11Les modèles de vision artificielle, comme le plus récent, prennent en charge une variété de tâches de vision artificielle telles que la détection d'objets, la classification d'images et la segmentation d'instances. Chacune de ces tâches vise à reproduire un aspect spécifique de la vision humaine, ce qui permet aux machines de voir et d'interpréter le monde qui les entoure. 

Par exemple, considère comment un élève en cours d'art peut prendre un crayon et tracer le contour d'un objet dans un dessin. En coulisse, son cerveau effectue une segmentation, c'est-à-dire qu'il distingue l'objet de l'arrière-plan et d'autres éléments. La segmentation des images atteint un objectif similaire en utilisant l'intelligence artificielle (IA), en décomposant les données visuelles en éléments significatifs pour que les machines puissent les comprendre. Cette technique peut être utilisée dans une variété d'applications dans de nombreux secteurs. 

Fig 1. Ultralytics YOLO11 utilisé pour segmenter les objets d'une image.

Un exemple pratique est la segmentation des pièces automobiles. En identifiant et en catégorisant les composants spécifiques d'un véhicule, la segmentation d'images peut rationaliser les processus dans des secteurs tels que la fabrication et la réparation d'automobiles, et le catalogage de commerce électronique.

Dans cet article, nous allons voir comment tu peux utiliser Ultralytics YOLO11 , Google Colab et l'ensemble de données Roboflow Carparts Segmentation pour créer une solution capable d'identifier et de segmenter avec précision les pièces de voiture.

Ultralytics YOLO11 est facile à utiliser

Ultralytics YOLO11 est disponible sous la forme d'un modèle pré-entraîné sur l'ensemble de données COCO, couvrant 80 classes d'objets différentes. Cependant, pour des applications spécifiques, telles que la segmentation de pièces de voiture, le modèle peut être entraîné sur mesure pour mieux s'adapter à ton ensemble de données et à ton cas d'utilisation. Cette flexibilité permet à YOLO11 d'être performant aussi bien dans les tâches générales que dans les tâches hautement spécialisées.

La formation personnalisée consiste à utiliser le modèle YOLO11 préformé et à l'affiner sur un nouvel ensemble de données. En fournissant des exemples étiquetés spécifiques à ta tâche, le modèle apprend à reconnaître et à segmenter les objets propres à ton projet. L'entraînement personnalisé garantit une précision et une pertinence accrues par rapport à l'utilisation de poids génériques pré-entraînés.

La configuration de YOLO11 pour la formation personnalisée est simple. Avec une configuration minimale, tu peux charger le modèle et l'ensemble de données, commencer la formation et surveiller les paramètres tels que la perte et la précision pendant le processus. YOLO11 comprend également des outils intégrés pour la validation et l'évaluation, ce qui facilite l'évaluation de la performance de ton modèle. 

Running Ultralytics YOLO11 sur Google Colab

Lors de la formation personnalisée YOLO11, il y a plusieurs options différentes pour mettre en place un environnement. L'un des choix les plus accessibles et les plus pratiques est Google Colab. Voici quelques avantages de l'utilisation de Google Colab pour la formation YOLO11 :

  • Accès gratuit aux ressources : Google Colab fournit des GPU (Graphics Processing Units) et des TPU (Tensor Processing Units), ce qui te permet de former YOLO11 sans matériel coûteux.
  • Environnement collaboratif : Google Colab t'aide à partager des carnets de notes, à stocker ton travail sur Google Drive et à simplifier le travail d'équipe en facilitant la collaboration et le suivi des versions.
  • Bibliothèques préinstallées : Avec des outils préinstallés tels que PyTorch et TensorFlow, Google Colab simplifie le processus d'installation et t'aide à démarrer rapidement.
  • Intégration dans le cloud : Tu peux facilement charger des ensembles de données depuis Google Drive, GitHub ou d'autres sources cloud, ce qui simplifie la préparation et le stockage des données.
Fig 2. Le carnet Google Colab YOLO11 .

Ultralytics propose également un ordinateur portableGoogle Colab préconfiguré, spécialement conçu pour la formation YOLO11 . Ce carnet comprend tout ce dont tu as besoin, de la formation au modèle à l'évaluation des performances, ce qui rend le processus simple et facile à suivre. C'est un excellent point de départ qui te permet de te concentrer sur l'adaptation du modèle à tes besoins spécifiques sans te soucier des étapes de configuration compliquées.

Un aperçu de l'ensemble de données de segmentation Carparts Roboflow

Après avoir décidé de ton environnement de formation, l'étape suivante consiste à rassembler des données ou à choisir un ensemble de données approprié pour segmenter les pièces de voiture. L'ensemble de données Roboflow Carparts Segmentation Dataset, disponible sur Roboflow Universe, est géré par Roboflow, une plateforme qui fournit des outils pour la construction, l'entraînement et le déploiement de modèles de vision par ordinateur. Cet ensemble de données comprend 3 156 images d'entraînement, 401 images de validation et 276 images de test, toutes avec des annotations de haute qualité pour les pièces de voiture comme les pare-chocs, les portes, les rétroviseurs et les roues.

Normalement, tu devrais télécharger l'ensemble de données à partir de Roboflow Universe et le configurer manuellement pour la formation sur Google Collab. Cependant, le paquetUltralytics Python simplifie ce processus en offrant une intégration transparente et des outils préconfigurés.

Fig 3. Exemples tirés de l'ensemble de données sur la segmentation des pièces de voiture.

Avec Ultralytics, l'ensemble de données est prêt à être utilisé grâce à un fichier YAML préconfiguré qui comprend les chemins d'accès à l'ensemble de données, les étiquettes de classe et d'autres paramètres de formation. Cela prend en charge la configuration pour toi, de sorte que tu peux rapidement charger l'ensemble de données et passer directement à l'entraînement de ton modèle. De plus, l'ensemble de données est structuré avec des ensembles d'entraînement, de validation et de test dédiés, ce qui facilite le suivi des progrès et l'évaluation des performances.

En exploitant l'ensemble de données de segmentation de Roboflow Carparts avec les outils fournis par Ultralytics YOLO11 , tu disposes d'un flux de travail transparent pour construire des modèles de segmentation de manière efficace sur des plateformes comme Google Colab. Cette approche réduit le temps de configuration et te permet de te concentrer sur l'affinement de ton modèle pour des applications réelles.

Applications réelles de la segmentation des pièces automobiles

La segmentation des pièces automobiles a une variété d'utilisations pratiques dans différents secteurs d'activité. Par exemple, dans les ateliers de réparation, elle peut aider à identifier et à catégoriser rapidement les composants endommagés afin de rendre le processus de réparation plus rapide et plus efficace. De même, dans le secteur des assurances, les modèles de segmentation peuvent automatiser l'évaluation des demandes d'indemnisation en analysant les images des véhicules endommagés pour identifier les pièces affectées. Cela permet d'accélérer le processus de réclamation, de réduire les erreurs et de faire gagner du temps aux assureurs et aux clients.

Fig 4. Segmentation des pièces de voiture à l'aide de YOLO.

En ce qui concerne la fabrication, la segmentation prend en charge le contrôle de la qualité en inspectant les pièces de voiture à la recherche de défauts, en assurant la cohérence et en réduisant les déchets. Ces applications montrent comment la segmentation des pièces automobiles peut transformer les industries en rendant les processus plus sûrs, plus rapides et plus précis.

Guide pas à pas : utilisation de YOLO11 sur Google Colab 

Maintenant que nous avons couvert tous les détails, il est temps de tout assembler. Pour commencer, tu peux consulter notre vidéo YouTube, qui te guide tout au long du processus de mise en place, de formation et de validation d'un modèle YOLO11 pour la segmentation des pièces automobiles.

Voici un aperçu rapide des étapes à suivre :

  • Configure ton environnement sur Google Colab: Active la prise en charge de GPU et installe le paquetage Ultralytics Python pour préparer la formation au modèle.
  • Charge le modèle YOLO11 : Commence par un modèle de segmentation YOLO11 pré-entraîné pour gagner du temps et exploiter les caractéristiques existantes pour la segmentation des pièces de voiture.
  • Entraîne le modèle avec l'ensemble de données: Utilise le fichier "carparts-seg.yaml" pendant l'entraînement pour télécharger, configurer et utiliser automatiquement l'ensemble de données de segmentation Carparts Roboflow . Ajuste les paramètres tels que les époques, la taille de l'image et la taille du lot pour affiner le modèle.
  • Surveille les progrès de la formation: Effectue le suivi des principales mesures de performance, telles que la perte de segmentation et la précision moyenne (mAP), pour t'assurer que le modèle s'améliore comme prévu.
  • Valider et déployer le modèle: Teste le modèle entraîné sur l'ensemble de validation pour confirmer sa précision et l'exporter pour des applications réelles comme le contrôle de la qualité ou le traitement des demandes d'assurance.

Avantages de l'utilisation de YOLO11 pour la segmentation des pièces automobiles

YOLO11 est un outil fiable et efficace pour la segmentation des pièces automobiles, offrant une série d'avantages qui le rendent idéal pour diverses applications dans le monde réel. En voici les principaux avantages :

  • Rapidité et efficacité : YOLO11 traite les images rapidement tout en conservant une grande précision, ce qui le rend adapté aux tâches en temps réel telles que le contrôle qualité et les véhicules autonomes.
  • Grande précision: Le modèle excelle dans la détection et la segmentation d'objets multiples au sein d'une même image, ce qui garantit une identification précise des pièces de voiture.
  • Évolutivité: YOLO11 peut gérer de grands ensembles de données et des tâches de segmentation complexes, ce qui le rend évolutif pour les applications industrielles.
  • Plusieurs intégrationsLe site Ultralytics prend en charge les intégrations avec des plateformes telles que Google Colab, Ultralytics Hub et d'autres outils populaires, ce qui améliore la flexibilité et l'accessibilité pour les développeurs.

Conseils pour travailler avec YOLO11 sur Google Collab

Bien que Google Colab rende les flux de travail d'apprentissage automatique beaucoup plus faciles à gérer, il peut falloir un peu de temps pour s'y habituer si tu es novice en la matière. Naviguer dans la configuration basée sur le cloud, les paramètres d'exécution et les limites de session peut sembler délicat au début, mais il existe quelques astuces qui peuvent rendre les choses beaucoup plus fluides.

Voici quelques considérations à prendre en compte :

  • Commence par activer l'accélération GPU dans les paramètres d'exécution pour accélérer l'entraînement. 
  • Comme Colab fonctionne dans le cloud, assure-toi d'avoir une connexion internet stable pour accéder aux ressources telles que les jeux de données et les référentiels. 
  • Organise tes fichiers et tes ensembles de données sur Google Drive ou GitHub pour les rendre faciles à charger et à gérer dans Colab.
  • Si tu rencontres des limitations de mémoire sur la partie gratuite de Colab, essaie de réduire la taille de l'image ou la taille du lot pendant l'entraînement. 
  • N'oublie pas de sauvegarder régulièrement ton modèle et tes résultats, car les sessions Colab ont des limites de temps, et tu ne veux pas perdre tes progrès. 

Réalise plus de choses avec YOLO11

Ultralytics YOLO11Combiné à des plateformes comme Google Colab et à des ensembles de données comme Roboflow Carparts Segmentation dataset, il rend la segmentation d'images simple et accessible. Avec ses outils intuitifs, ses modèles pré-entraînés et sa configuration facile, YOLO11 te permet de te plonger facilement dans des tâches avancées de vision par ordinateur. 

Que tu améliores la sécurité automobile, que tu optimises la fabrication ou que tu construises des applications d'IA innovantes, cette combinaison fournit les outils qui t'aideront à réussir. Avec Ultralytics YOLO11 , tu ne te contentes pas de construire des modèles - tu ouvres la voie à des solutions plus intelligentes et plus efficaces dans le monde réel.

Pour en savoir plus, consulte notre dépôt GitHub et engage-toi auprès de notre communauté. Explore les applications de l'IA dans les voitures autonomes et la vision par ordinateur pour l'agriculture sur nos pages de solutions. 🚀

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.