Yolo Vision Shenzhen
Shenzhen
Rejoindre maintenant

Une plongée en profondeur dans la détection des pilules à l'aide d'Ultralytics YOLO11

Abirami Vina

5 min de lecture

24 avril 2025

Dans ce tutoriel de codage, apprenez à detect pilules à l'aide de YOLO11 avec l'ensemble de données Medical-Pills. Explorez également ses applications et avantages potentiels.

L'intelligence artificielle est utilisée dans presque tous les secteurs, mais son influence sur la santé, en particulier dans le domaine pharmaceutique, est particulièrement importante. Cette année, le marché de l'IA dans le secteur pharmaceutique est évalué à 1,94 milliard de dollars, et il devrait atteindre 16,49 milliards de dollars d'ici 2034. 

Un moteur technologique clé de ce marché est la vision par ordinateur. Contrairement au traitement d'image traditionnel, la vision par ordinateur est un sous-domaine de l'IA qui permet aux machines de comprendre et d'analyser les données visuelles en temps réel. 

Fig 1. Un aperçu de l'IA sur le marché pharmaceutique.

Dans l'industrie pharmaceutique, où même la plus petite erreur peut avoir de graves conséquences, la Vision IA offre de nouvelles façons fiables d'améliorer la sécurité et la précision.  

Par exemple, Ultralytics YOLO11 est un modèle de vision par ordinateur conçu pour des tâches en temps réel telles que la détection d'objets et la segmentation d'instances. Il peut être utilisé pour des applications telles que l'identification de pilules ou la détection de défauts dans les emballages médicaux. 

Dans cet article, nous allons voir comment mettre en pratique Vision AI en entraînant YOLO11 à detect pilules. Nous explorerons également ses applications dans le monde réel. Commençons par le commencement !

Création d'un modèle d'IA de détection de pilules

Avant de voir comment YOLO11 peut être entraîné à detect pilules, prenons un peu de recul pour comprendre ce que signifie l'entraînement d'un modèle et le rôle d'un ensemble de données. 

L'entraînement d'un modèle consiste à lui apprendre à reconnaître des motifs en lui montrant de nombreux exemples. Dans ce cas, un ensemble de données est une collection d'images où chaque image est étiquetée pour indiquer où se trouvent les pilules. Ce processus aide le modèle à apprendre à partir de ces exemples afin qu'il puisse ensuite identifier les pilules dans de nouvelles images.

L'ensembleUltralytics Python rend ce processus encore plus facile en prenant en charge un large éventail d'ensembles de données dans un format de fichier YOLO rationalisé. Il offre un accès aisé aux ensembles de données les plus courants et prend en charge des applications telles que la détection de pilules. 

Par exemple, l'ensemble de données Medical Pills est une collection dédiée de validation de concept conçue pour montrer comment la détection d'objets peut améliorer les flux de travail pharmaceutiques grâce à des tâches telles que le contrôle de la qualité, le tri et la détection des contrefaçons.

Choisir un environnement de développement

Un autre facteur à prendre en compte avant de commencer la formation avec le package Ultralytics Python est le choix de l'environnement de développement adéquat. Voici trois options populaires :

  • Interface de ligne de commande (CLI) : L'interface de CLI de commande ou terminal est un outil simple, basé sur du texte, dans lequel vous pouvez saisir des commandes pour exécuter votre code et interagir avec votre ordinateur.

  • Notebooks Jupyter : Il s'agit d'un environnement plus interactif où vous pouvez écrire et exécuter du code en petits blocs (cellules), ce qui facilite les tests et le débogage au fur et à mesure.

  • Google Colab : Une option basée sur le cloud qui fonctionne comme Jupyter Notebooks, mais avec en prime un accès gratuit au GPU , de sorte que vous n'avez pas à vous soucier de configurer quoi que ce soit localement.

Il existe d'autres options de configuration que vous pouvez explorer dans la documentation officielle d'Ultralytics , mais les trois mentionnées ci-dessus sont faciles à configurer et à utiliser, ce qui en fait d'excellents choix pour démarrer rapidement.

Dans ce guide, nous nous concentrerons sur la manière de configurer et d'entraîner YOLO11 en utilisant Google Colab, Jupyter Notebooks ou un script Python de base, car le processus est assez similaire dans chacun de ces environnements.

Par ailleurs, ce tutoriel est assez similaire à celui que nous avons couvert précédemment sur la détection de la faune et de la flore à l'aide de YOLO11. Si vous souhaitez obtenir plus de détails sur l'une des étapes de ce tutoriel de codage, vous pouvez le consulter.

Exploration d'un ensemble de données pour la détection de pilules en temps réel avec YOLO

L'ensemble de données Medical Pills comprend 92 images d'entraînement et 23 images de validation, offrant une division solide pour la construction et le test de votre modèle. Les images d'entraînement sont utilisées pour enseigner au modèle, tandis que les images de validation aident à évaluer les performances du modèle sur des données nouvelles et non vues. 

Chaque image de l'ensemble de données est étiquetée pour une seule classe, les pilules. Les annotations de la boîte englobante marquent clairement l'emplacement de chaque pilule, ce qui rend l'ensemble de données idéal pour les tâches ciblées telles que la détection de pilules sans la complexité de la gestion de plusieurs classes d'objets.

Fig. 2. Un aperçu du jeu de données Medical Pills.

Pour soutenir l'entraînement avec YOLO11, Ultralytics fournit un fichier de configuration YAML qui définit les paramètres clés tels que les chemins d'accès aux fichiers, les noms de classe et les métadonnées nécessaires à l'entraînement du modèle. Que vous affiniez un modèle pré-entraîné ou que vous partiez de zéro, ce fichier simplifie grandement le processus et vous aide à démarrer rapidement.

Entraîner un modèle Ultralytics YOLO sur le jeu de données des pilules

Pour commencer, nous allons mettre en place un environnement pour l'entraînement et le test du modèle. Vous pouvez choisir d'utiliser Google Colab, les carnets Jupyter ou un simple fichier Python , selon vos préférences. Il vous suffit de créer un nouveau carnet de notes ou un fichier Python dans l'environnement de votre choix.

Ensuite, nous pouvons configurer notre environnement et installer le paquetage Ultralytics Python à l'aide de la commande ci-dessous. Si vous utilisez un environnement basé sur un carnet de notesGoogle Colab ou Jupyter), exécutez la commande suivante avec un point d'exclamation ( !) au début.

pip install ultralytics

Une fois installé, l'étape suivante consiste à télécharger et à entraîner YOLO11 à l'aide de l'ensemble de données Medical Pills. Comme ce jeu de données est pris en charge par le paquetagePython Ultralytics , la procédure est simple. 

Comprendre le processus d'entraînement du modèle

Tout d'abord, nous pouvons importer la classe YOLO à partir du paquet Ultralytics . Ensuite, nous pouvons charger un modèle YOLO11 pré-entraîné à partir du fichier "yolo11n.pt", qui est recommandé parce qu'il s'agit d'un modèle nano et léger. 

Enfin, nous pouvons lancer le processus d'apprentissage en orientant le modèle vers la configuration de notre jeu de données (medical-pillsyaml) et en fixant le nombre d'époques d'apprentissage (un passage complet sur l'ensemble du jeu de données) à 100, comme indiqué ci-dessous.

from ultralytics import YOLO

model = YOLO("yolo11n.pt")

results = model.train(data="medical-pills.yaml", epochs=100)

L'entraînement sur plusieurs epochs permet au modèle d'apprendre et d'améliorer ses performances à chaque passage. Vous pourrez trouver les journaux et les points de contrôle enregistrés dans le sous-dossier “runs/train/”, que vous pourrez utiliser pour suivre les progrès et examiner les performances du modèle.

Une fois l'entraînement terminé, le modèle YOLO11 entraîné sur mesure devrait être en mesure d'identifier les pilules avec précision. Vous trouverez les poids finaux du modèle entraîné dans le sous-dossier "detect" sous le nom "best.pt".

Évaluation de YOLO11 après l'apprentissage du modèle

Pour évaluer dans quelle mesure le modèle a appris à detect pilules, nous pouvons procéder à une validation comme suit :

metrics = model.val()

Ce processus renvoie des métriques courantes de détection d'objets, qui donnent un aperçu des performances du modèle. Voici un aperçu de certaines de ces métriques :

  • Précision : Elle mesure la proportion de pilules détectées par le modèle qui sont correctes.
  • Rappel : Il indique la proportion de pilules réelles que le modèle identifie avec succès.
  • Précision moyenne (mAP) : Cette mesure combine à la fois la précision et le rappel pour différents seuils de détection afin d'obtenir une note de performance globale.

Ensemble, ces métriques offrent une vue d'ensemble de la précision avec laquelle le modèle détecte les pilules dans de nouvelles données non observées. 

Si votre modèle ne fonctionne pas aussi bien que prévu, vous pouvez essayer de l'entraîner pendant plus d'epochs ou d'affiner d'autres paramètres d'entraînement, tels que le taux d'apprentissage, qui contrôle la taille des pas effectués lors de l'optimisation du modèle, ou la taille de l'image, afin d'améliorer encore ses performances.

Effectuer des inférences à l'aide de votre modèle YOLO11 personnalisé

Une fois le modèle YOLO11 formé et évalué, l'étape suivante consiste à tester ses performances sur de nouvelles images inédites. Cela permet de simuler des conditions réelles, telles que la détection de pilules sous différents éclairages, dispositions ou styles d'emballage.

Pour tester le modèle, nous avons téléchargé un exemple d'image à partir de Pexels, un site web d'images gratuites, et nous avons analysé l'image ou effectué une prédiction en utilisant le modèle YOLO11 entraîné sur mesure, comme le montre l'extrait de code ci-dessous. 

Vous pouvez utiliser cet exemple d'image ou toute autre image pertinente pour évaluer les performances du modèle dans des scénarios réels. 

results = model.predict("path/to/image.jpg", save=True, conf=0.3)

L'option de sauvegarde indique au modèle de stocker l'image de sortie, et le paramètre de confiance garantit que seules les prédictions avec au moins 30 % de certitude sont incluses dans les résultats.

Lorsque vous exécutez la prédiction, la sortie affiche un message indiquant l'emplacement de l'image enregistrée - par exemple, "Results saved to detect" (résultats enregistrés dans runs/detect/train).

Votre image de sortie sera similaire à celle illustrée ici, avec des pilules détectées et mises en évidence à l'aide de cadres de délimitation. Les scores de confiance affichés indiquent le niveau de certitude pour chaque détection.

Fig. 3. Détection de pilules à l'aide de YOLO11.

Applications concrètes de l'utilisation de YOLO11 dans l'industrie pharmaceutique

Maintenant que nous avons exploré la manière d'entraîner YOLO11 à l'aide de l'ensemble de données Medical-Pills et d'exécuter des inférences sur des images pour la détection de pilules, examinons les applications réelles de YOLO11dans l'industrie pharmaceutique.

Tri de pilules pharmaceutiques avec YOLO11

La détection automatisée de pilules avec YOLO11 peut être appliquée au triage des produits pharmaceutiques. Le tri manuel est souvent lent, répétitif et sujet à des erreurs qui peuvent compromettre la sécurité et la conformité des médicaments. 

En utilisant un modèle YOLO11 affiné, nous pouvons detect et trier avec précision les pilules sur la base d'attributs visuels tels que la taille, la forme et la couleur. Cette automatisation accélère le processus et contribue à garantir que les produits répondent à des normes de qualité strictes, ce qui en fait un outil précieux pour les opérations pharmaceutiques.

Fig. 4. Détection de pilules à l'aide de YOLO11.

Suivi de l'inventaire à l'aide de YOLO11

Avoir les bons médicaments en stock à temps est plus qu'une simple tâche logistique - cela peut affecter les soins aux patients et les coûts. Une pénurie d'un comprimé essentiel peut retarder le traitement, tandis qu'un stockage excessif peut entraîner l'expiration des médicaments et le gaspillage des stocks. Avec de nombreux types de comprimés et de variations d'emballage dans l'industrie pharmaceutique, les systèmes d'inventaire automatisés peuvent permettre des enregistrements plus précis.

Les systèmes d'inventaire intelligents peuvent utiliser des modèles de vision artificielle comme Ultralytics YOLO11 pour contrôler les niveaux de stock en temps réel. Le modèle peut scanner les étagères et les zones d'emballage à l'aide d'images ou de vidéos pour detect et compter les pilules. Lorsque les niveaux de stock changent, que des articles soient ajoutés, retirés ou déplacés, le système peut mettre à jour le comptage automatiquement.

Le contrôle de la qualité des produits pharmaceutiques piloté par YOLO11

Dans la production pharmaceutique, le contrôle de la qualité est crucial pour s'assurer que chaque pilule est sûre et efficace. Même des défauts mineurs, tels qu'une fissure, une forme irrégulière ou une légère variation de couleur, peuvent entraîner des erreurs de dosage ou des rappels de produits. 

YOLO11 peut vous aider en détectant automatiquement les pilules qui ne répondent pas aux normes de qualité. Le modèle peut apprendre les caractéristiques visuelles et utilise des boîtes de délimitation pour signaler en temps réel les problèmes tels que les ébréchures, les empreintes décolorées ou la décoloration. Cela permet de retirer rapidement les pilules défectueuses, de réduire les déchets et de garantir que seuls les médicaments dont la qualité est assurée parviennent aux patients.

En outre, YOLO11 peut être utilisé pour detect et compter les pilules au fur et à mesure qu'elles sont inspectées, pour un suivi précis tout en contrôlant la qualité. 

Fig. 5. YOLO11 peut être utilisé pour detect et compter les capsules.

Avantages et inconvénients de l'utilisation de la Vision IA pour la détection de pilules 

Maintenant que nous avons exploré comment la Vision IA peut être appliquée dans l'industrie pharmaceutique, examinons rapidement quelques-uns des principaux avantages de l'utilisation de la vision par ordinateur dans ce secteur :

  • Maintenance prédictive : YOLO11 peut être utilisé pour detect signes précoces d'usure des machines en identifiant les incohérences de pilules ou d'emballage. Il permet de programmer des réparations en temps voulu et d'éviter les arrêts de production imprévus.
  • Utilisation évolutive du modèle : Le modèle peut être affiné sur divers ensembles de données pour différents comprimés et emballages. Cela rend l'inspection évolutive et rentable à mesure que les opérations se développent.
  • Surveillance à distance : Il permet des contrôles de qualité en temps réel lorsqu'il est intégré à des systèmes cloud et à des appareils périphériques et est idéal pour la gestion des distributeurs ruraux, des unités automatisées et des installations de télépharmacie à distance.

Bien qu'il existe de nombreux avantages à mettre en œuvre l'IA de vision dans l'industrie pharmaceutique, il y a également quelques considérations à garder à l'esprit lors de l'utilisation de ces technologies : 

  • Intégration opérationnelle : L'intégration de systèmes d'IA dans les flux de travail existants peut nécessiter des ajustements, une formation et des vérifications de compatibilité avec l'infrastructure actuelle.
  • Conformité réglementaire: Les systèmes automatisés doivent respecter des normes réglementaires strictes pour garantir la sécurité des patients et la qualité constante des produits.
  • Gestion des erreurs : Même les modèles avancés peuvent produire des faux positifs ou des faux négatifs. Il est important de mettre en place des processus pour gérer et corriger ces erreurs.

La voie à suivre pour les flux de travail de l'IA pharmaceutique

À l'avenir, l'IA jouera probablement un rôle plus important dans l'accélération, l'amélioration et la réduction des coûts des essais cliniques. Elle peut aider à concevoir de meilleurs protocoles d'essai, à choisir les bons groupes de patients et à surveiller les données en temps réel. 

Cela peut permettre aux chercheurs de répondre aux problèmes au fur et à mesure qu'ils surviennent, au lieu de le faire après coup. L'IA peut également accélérer le processus d'approbation en réduisant la paperasserie manuelle et en automatisant les contrôles de routine. Dans l'ensemble, l'intégration de l'IA dans les flux de travail pharmaceutiques peut entraîner moins de retards et un accès plus rapide aux nouveaux traitements.

Principaux points à retenir

L'entraînement d'Ultralytics YOLO11 sur l'ensemble de données Medical Pills montre la rapidité et l'efficacité avec lesquelles le modèle peut s'adapter aux tâches pharmaceutiques. Même avec un petit ensemble de données, il peut detect avec précision detect pilules, ce qui le rend utile pour des tâches telles que le tri, le contrôle de la qualité et le suivi des stocks.

À mesure que les ensembles de données augmentent et que les modèles s'améliorent, le potentiel de la Vision IA dans le secteur pharmaceutique va au-delà de la simple logistique. Cette technologie pourrait également soutenir les essais cliniques en aidant à l'identification et au suivi cohérents des pilules, et en aidant les chercheurs à tester en toute sécurité de nouvelles combinaisons de médicaments. 

Explorez notre dépôt GitHub pour en savoir plus et faire partie de notre communauté en pleine croissance. Découvrez des innovations de pointe dans divers secteurs, de l'IA dans l'agriculture à la vision par ordinateur dans les soins de santé. Consultez nos options de licence et lancez vos projets Vision AI dès aujourd'hui.

Construisons ensemble l'avenir
de l'IA !

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

Démarrer gratuitement