Comment utiliser Ultralytics YOLO11 pour la classification d'images

Imaginons qu'un robot observe deux chats, un noir et un blanc, et doive déterminer lequel est lequel. Pour ce faire, il peut utiliser la classification d'images, une tâche de vision par ordinateur qui aide à identifier et à catégoriser des objets ou des scènes dans une image. En fait, grâce aux récentes avancées en intelligence artificielle (IA), la classification d'images peut être utilisée dans une grande variété d'applications allant de la surveillance animale à la fabrication et à l'agriculture avec la détection des maladies des cultures.

L'une des dernières avancées en matière de classification d'images est le modèle Ultralytics YOLO11. Lancé lors de l'événement hybride annuel d'Ultralytics, YOLO Vision 2024 (YV24), YOLO11 est conçu pour traiter une grande variété de tâches de vision par IA avec facilité et précision, y compris la classification d'images.

Dans cet article, nous explorerons les fondamentaux de la classification d'images, discuterons d'applications concrètes et te montrerons comment utiliser YOLO11 pour la classification d'images via le package Python Ultralytics. Nous examinerons également comment essayer les capacités de YOLO11 sur Ultralytics HUB en quelques étapes simples. Commençons !

Fig 1. Un exemple d'utilisation d'Ultralytics YOLO11 pour classer un chat persan.

Link to this sectionQu'est-ce que la classification d'images ?#

La classification d'images fonctionne en attribuant une étiquette ou une balise à une image basée sur des modèles appris à partir d'images précédemment étiquetées. En analysant soigneusement les pixels d'une image, un modèle de vision par ordinateur peut trouver la meilleure correspondance pour l'image. Des modèles fiables comme YOLO11 peuvent gérer ce processus de manière transparente. L'architecture de modèle de YOLO11 permet de traiter des images ou des images vidéo presque instantanément, ce qui le rend idéal pour les applications nécessitant une classification d'images rapide et précise.

Pour vraiment comprendre la portée de la classification d'images, il est utile de la distinguer d'autres tâches comme la détection d'objets. Alors que la classification d'images étiquette une image entière, la détection d'objets identifie et localise chaque objet au sein de l'image.

Fig 2. Une comparaison de la classification d'images, de la détection d'objets et de la segmentation d'images.

Prenons l'image d'une girafe. En classification d'images, le modèle pourrait étiqueter l'image entière simplement comme étant une girafe en fonction de son contenu global. Cependant, avec la détection d'objets, le modèle ne s'arrête pas à l'identification de la girafe ; il place également une boîte englobante autour d'elle, identifiant son emplacement exact dans l'image.

Maintenant, imagine la girafe debout près d'un arbre dans une savane avec d'autres animaux. Un modèle de classification d'images pourrait étiqueter toute la scène comme étant une savane ou simplement de la faune. Cependant, avec la détection d'objets, le modèle identifierait chaque élément individuellement, reconnaissant la girafe, l'arbre et les autres animaux, chacun avec ses propres boîtes englobantes.

Link to this sectionApplications de la classification d'images YOLO11#

La précision et la performance du modèle Ultralytics YOLO11 pour la classification d'images le rendent utile dans un large éventail d'industries. Explorons certaines des applications clés de YOLO11 dans la classification d'images.

Link to this sectionLa classification d'images YOLO11 dans l'agriculture#

La classification d'images peut aider à rationaliser de nombreuses fonctions dans l'industrie de l'agriculture et de l'élevage. Plus précisément, en utilisant des modèles de classification d'images comme YOLO11, les agriculteurs peuvent surveiller en permanence la santé de leurs cultures, détecter des maladies graves et identifier toute infestation de nuisibles avec une grande précision.

Voici un aperçu de la façon dont cela fonctionne :

• Capture d'image : Des appareils Internet des objets (IoT) comme des caméras et des drones peuvent être déployés pour capturer des images en temps réel des cultures sous divers angles et emplacements dans les champs.
• Traitement : En fonction des ressources disponibles et de la connectivité, les images peuvent être traitées sur place via edge computing ou téléchargées sur le cloud pour une analyse plus intensive.
• Classification d'images avec YOLO11 : Le modèle YOLO11 peut analyser ces images pour classifier diverses conditions des cultures. Les classes courantes peuvent inclure sain, malade, infesté par des nuisibles ou carencé en nutriments, aidant à identifier des problèmes spécifiques affectant différentes zones du champ.
• Génération d'informations : Sur la base des classifications, YOLO11 fournit des informations sur les indicateurs de santé des cultures, aidant les agriculteurs à détecter les premiers signes de maladie, à identifier les points chauds de nuisibles ou à repérer les carences en nutriments.
• Prise de décision éclairée : Avec ces informations, les agriculteurs peuvent prendre des décisions ciblées sur l'irrigation, la fertilisation et le contrôle des nuisibles, en n'appliquant les ressources que là où elles sont le plus nécessaires.

Fig 3. Un exemple de différentes classes de feuilles, de saines à infectées.

Link to this sectionLa classification d'images YOLO11 dans le commerce de détail#

La classification d'images peut améliorer considérablement l'expérience d'achat au détail, en la rendant plus personnalisée et conviviale. Les détaillants peuvent utiliser des modèles de vision par ordinateur entraînés sur mesure pour reconnaître les produits dans leur inventaire et intégrer cette capacité dans leurs applications mobiles ou sites web. Les clients peuvent alors rechercher des produits simplement en téléchargeant une photo, rendant les achats plus rapides et plus pratiques.

Une fois qu'un client télécharge une image dans un système de recherche visuelle, plusieurs choses se produisent en coulisses avant que les résultats de recherche ne s'affichent.

Tout d'abord, la détection d'objets peut être utilisée pour sélectionner les éléments principaux de l'image, comme l'identification d'un vêtement ou d'un meuble et sa séparation de l'arrière-plan. Ensuite, la classification d'images peut être utilisée pour catégoriser davantage chaque élément, en reconnaissant s'il s'agit d'une veste, d'une chemise, d'un canapé ou d'une table.

Avec ces informations, le système peut afficher des produits similaires disponibles à l'achat, ce qui est particulièrement utile pour trouver des articles uniques ou tendance difficiles à décrire avec des mots seuls. La même technologie peut également aider à rationaliser d'autres tâches de détail, comme la gestion des stocks, en reconnaissant et en catégorisant automatiquement les articles.

Fig 4. Une plateforme de recherche visuelle basée sur la classification d'images en action.

Link to this sectionSurveillance de la faune avec la classification d'images YOLO11#

Traditionnellement, surveiller les animaux dans la nature est une tâche fastidieuse impliquant de nombreuses personnes triant et analysant manuellement des milliers de photos. Avec des modèles de vision par ordinateur comme YOLO11, les chercheurs peuvent surveiller automatiquement les animaux à un rythme plus rapide. Des caméras peuvent être placées dans les habitats naturels pour prendre des photos. Le modèle de vision par IA peut ensuite être utilisé pour analyser ces photos et classifier les animaux qu'elles contiennent (le cas échéant). Un tel système peut aider les chercheurs à étudier et à suivre les populations animales, leurs modèles de migration, etc.

Une autre façon dont les modèles d'IA et de vision par ordinateur comme YOLO11 peuvent aider dans ce domaine est en rationalisant le processus de classification des espèces menacées. En identifiant les espèces potentielles ou les catégories de races auxquelles un animal peut appartenir, ces modèles peuvent fournir des données essentielles aux chercheurs. Par exemple, l'Université de Tasmanie (UTAS) a développé un système basé sur la classification d'images pour surveiller différentes espèces sauvages de Tasmanie. Les prédictions des modèles peuvent ensuite aider les scientifiques et les chercheurs à surveiller l'activité et le comportement des animaux, ce qui peut signaler des menaces comme le braconnage ou la perte d'habitat.

Fig 5. YOLO11 prédisant les races possibles auxquelles un chien pourrait appartenir.

Link to this sectionEssayer la classification d'images avec le modèle YOLO11#

Maintenant que nous avons discuté de ce qu'est la classification d'images et exploré certaines de ses applications, voyons comment tu peux essayer la classification d'images avec le nouveau modèle YOLO11. Il existe deux façons simples de commencer : en utilisant le package Python Ultralytics ou via Ultralytics HUB. Nous allons parcourir les deux options.

Link to this sectionExécuter des inférences avec YOLO11#

Pour commencer avec le package Python Ultralytics, installe-le simplement en utilisant pip, conda ou Docker. Si tu rencontres des problèmes, consulte notre Guide des problèmes courants pour obtenir des conseils de dépannage utiles.

Une fois le package installé, tu peux utiliser le code suivant pour charger une variante du modèle de classification d'images YOLO11 et exécuter une inférence sur une image. Exécuter une inférence signifie utiliser un modèle entraîné pour faire des prédictions sur de nouvelles données inédites. Tu peux essayer avec l'image de ton choix !

Fig 6. Exécution d'inférences à l'aide du modèle YOLO11.

Link to this sectionEntraîner un modèle de classification YOLO11 personnalisé#

Tu peux également utiliser le même package Python pour entraîner un modèle de classification YOLO11 personnalisé. L'entraînement personnalisé te permet de peaufiner un modèle YOLO11 pour tes besoins spécifiques. Par exemple, si tu développes une application pour classifier différentes races de chats, tu peux entraîner sur mesure un modèle YOLO11 juste pour cet objectif.

Le code ci-dessous montre comment charger et entraîner un modèle de classification d'images YOLO11. Il te permet de transférer des poids pré-entraînés, en utilisant les connaissances d'un modèle existant pour améliorer les performances de ton propre modèle. Tu peux spécifier un jeu de données, comme le jeu de données « fashion-mnist », qui est un ensemble bien connu d'images en niveaux de gris d'articles vestimentaires (chemises, pantalons, chaussures, etc.). Entraîner le modèle sur ce jeu de données lui apprend à reconnaître différentes catégories de vêtements. Tu peux remplacer « fashion-mnist » par n'importe quel jeu de données qui convient à ton projet, comme des races de chats ou des types de plantes.

Fig 7. Entraînement personnalisé d'un modèle YOLO11 pour la classification d'images.

Link to this sectionEssayer YOLO11 sur Ultralytics HUB#

Bien que l'utilisation du package Ultralytics soit simple, elle nécessite quelques connaissances en Python. Si tu cherches une option plus conviviale pour les débutants, tu peux utiliser Ultralytics HUB, une plateforme conçue pour rendre l'entraînement et le déploiement de différents modèles YOLO simples et accessibles. Pour commencer, tu devras créer un compte.

Une fois connecté, navigue vers la section « Modèles » et sélectionne le modèle YOLO11 pour la classification d'images. Tu verras une gamme de tailles de modèles disponibles : nano, small, medium, large et extra-large. Après avoir choisi un modèle, tu peux télécharger une image dans la section « Aperçu », où les prédictions apparaîtront sur le côté gauche de la page une fois l'image traitée.

Fig 8. Utilisation d'Ultralytics HUB pour exécuter une inférence.

Link to this sectionPoints clés#

YOLO11 offre des capacités de classification d'images puissantes qui ouvrent de nouvelles possibilités dans divers secteurs. De l'amélioration de la surveillance des cultures dans l'agriculture à l'amélioration des recherches de produits dans le commerce de détail en passant par le soutien à la conservation de la faune, la vitesse et la précision de YOLO11 le rendent idéal pour des applications diverses. Avec des options pour un entraînement personnalisé via le package Python Ultralytics ou une configuration sans code conviviale sur Ultralytics HUB, les utilisateurs peuvent facilement intégrer YOLO11 dans leurs flux de travail. Alors que de plus en plus d'industries adoptent des solutions d'IA, YOLO11 offre un outil flexible et performant qui soutient l'innovation et les avancées pratiques.

Pour en explorer davantage, visite notre dépôt GitHub et engage-toi avec notre communauté. Explore les applications de l'IA dans les voitures autonomes et les soins de santé sur nos pages de solutions. 🚀