Exporter un modèle Ultralytics YOLOv8 formé sur mesure

Rejoignez-nous pour examiner de plus près la détection d'objets personnalisés avec Ultralytics YOLOv8. Dans ce billet, nous explorerons le processus complexe de formation d'un modèle personnalisé, l'exportation des poids formés et l'exécution d'une inférence en direct sur une webcam. 

Formation d'un modèle de détection d'objets personnalisé

Dans notre précédente vidéo, nous nous sommes penchés sur l' entraînement d' un modèle YOLOv8 sur un ensemble de données de coupes personnalisées dans Google Colab. Nous avons vu le graphique d'entraînement s'améliorer régulièrement, la perte diminuant et la position d'erreur moyenne augmentant. Sur cette base, notre modèle spécial peut désormais identifier très précisément cinq types de gobelets différents.

‍

Notre modèle personnalisé étant formé et prêt à fonctionner, il est temps d'explorer la prochaine frontière : exporter les poids formés et exécuter l'inférence en direct sur une webcam.

L'inférence de modèle est le processus d'utilisation d'un modèle de vision par ordinateur entraîné pour faire des prédictions ou prendre des décisions sur la base de nouvelles données inédites. C'est lorsque le modèle utilise des données d'entrée, telles qu'une image, et les traite à l'aide des paramètres et de la structure qu'il a appris. Le modèle produit ensuite un résultat, tel que la classification, la détection ou la segmentation, sur la base de sa tâche d'apprentissage. 

En termes pratiques, l'inférence implique souvent le déploiement du modèle formé dans un environnement de production où il peut être utilisé pour traiter des données réelles en temps réel ou dans des scénarios de traitement par lots.

Exporter et exécuter l'inférence avec le modèle formé

Une fois les poids du modèle entraînés téléchargés depuis Colab, nous pouvons les importer en toute transparence dans un environnement Python, prêts à libérer tout le potentiel de notre modèle personnalisé.

En quelques lignes de code, nous pouvons configurer un script Python pour lancer une inférence en direct sur une webcam, capturer des séquences en temps réel et détecter différentes tasses avec une précision impressionnante. La puissance de YOLOv8 transparaît lorsque notre modèle identifie sans effort des gobelets de formes, de tailles et de couleurs différentes, démontrant ainsi sa polyvalence et sa fiabilité dans des scénarios réels.

L'affinement des modèles présente plusieurs avantages. Ils peuvent détecter, segmenter ou classer des objets que les modèles pré-entraînés ne peuvent pas prendre en charge. En outre, ils peuvent aider les chercheurs ou les scientifiques des données à comprendre comment l'architecture du modèle fonctionne sur des ensembles de données réels.

Conclusion

De l'entraînement d'un modèle personnalisé à l'exportation des poids entraînés, en passant par l'inférence en direct sur une webcam, nous avons pu constater de visu la puissance et la polyvalence de YOLOv8.

Rejoignez-nous dans notre quête pour libérer le plein potentiel de la détection d'objets personnalisés avec Ultralytics YOLOv8. Ensemble, façonnons l'avenir de l'IA, une détection à la fois. Consultez le HUB Ultralytics et notre documentation pour plus d'informations sur tout ce qui concerne Ultralytics et rejoignez notre communauté pour vous tenir au courant des derniers développements ! 

Si vous êtes curieux de savoir comment exporter des modèles YOLOv8 formés sur mesure et comment faire de l'inférence sur une webcam, regardez la vidéo complète ici!