Comment entraîner tes propres modèles avec Ultralytics HUB

Alors que nous visons constamment l'innovation par l'IA, rejoins-nous pour examiner de plus près Ultralytics HUB, une plateforme conçue pour simplifier l'entraînement de modèles Ultralytics YOLO personnalisés. Que tu sois un développeur expérimenté ou débutant, Ultralytics HUB offre une expérience fluide pour créer et gérer des modèles de vision par ordinateur. Découvrons comment entraîner tes modèles YOLO personnalisés en quelques clics.

Link to this sectionQu'est-ce qu'Ultralytics HUB ?#

Ultralytics HUB est une plateforme complète conçue pour gérer tes jeux de données, tes projets et tes modèles, facilitant ainsi l'entraînement et le déploiement de modèles de machine learning.

Elle s'intègre à diverses plateformes, prend en charge les mises à jour en temps réel via son application mobile (disponible sur Android et iOS), et évolue continuellement avec de nouvelles fonctionnalités. Pour plus d'informations techniques et un flux de travail étape par étape, consulte notre article Medium sur la façon d'entraîner des modèles de vision par ordinateur dans le cloud.

Link to this sectionBien démarrer avec Ultralytics HUB#

Link to this sectionInterface conviviale#

La première chose que tu remarqueras à propos d'Ultralytics HUB est son interface intuitive. La page d'accueil offre un accès facile aux datasets, aux projets, aux modèles et aux intégrations. Bien qu'elle soit encore en version bêta, la plateforme dispose déjà d'une gamme de fonctionnalités puissantes.

Link to this sectionDatasets et modèles#

Ultralytics HUB est livré pré-chargé avec des datasets standard comme VOC, COCO et Simpsons, couramment utilisés pour le benchmarking. Ces datasets sont substantiels, COCO comprenant 80 classes et 140 000 images. Cependant, la plateforme prend également en charge des datasets plus petits, ce qui la rend idéale pour les utilisateurs de tous niveaux. Tu peux télécharger ton propre dataset personnalisé ou te connecter à des outils externes comme Roboflow pour l'annotation d'images et la préparation des données.

Link to this sectionCréer un projet#

Créer un nouveau projet dans Ultralytics HUB est incroyablement simple. Voici un guide étape par étape pour t'aider à démarrer :

1. Créer un projet : Commence par nommer ton projet. Pour cet exemple, appelons-le "Object Detection".
2. Choisir un dataset : Sélectionne un dataset parmi les options disponibles. Par souci de simplicité, nous utiliserons le dataset Simpsons avec 14 000 images.
3. Sélectionner un modèle : Choisis un modèle YOLO. Ultralytics HUB prend en charge divers modèles, y compris YOLOv5 et YOLOv8. Pour cet exemple, nous utiliserons le modèle nano de YOLOv8.
4. Ajuster les hyperparamètres : Modifie des paramètres tels que le nombre d'époques, la taille de l'image et la taille des lots (batch size). Tu peux également choisir d'utiliser un GPU ou un CPU et configurer les options de mise en cache.

Link to this sectionEntraîner le modèle#

Une fois ton projet configuré, l'entraînement du modèle se fait en un clic. Ultralytics HUB propose plusieurs options d'entraînement :

• Entraînement local : Installe Ultralytics sur ta machine locale et exécute le script d'entraînement.
• Google Colab : Ouvre un notebook Google Colab pré-configuré avec le code nécessaire. Cette option ne nécessite aucun codage et s'exécute de manière transparente dans le cloud.
• HUB Cloud : L'entraînement dans le cloud avec Ultralytics HUB offre une solution sans code pour entraîner des modèles YOLO, parfaite pour les non-codeurs et les chefs d'entreprise. Le flux de travail inclut le téléchargement du dataset, la sélection du modèle et la configuration de l'instance cloud, ce qui facilite le fine-tuning de modèles pré-entraînés et leur exportation pour diverses applications.

Pour entraîner le modèle dans Google Colab :

1. Installer Ultralytics : Exécute la commande pour installer Ultralytics dans le notebook.
2. Configuration et clé API : Configure l'installation et saisis ta clé API.
3. Démarrer l'entraînement : Exécute les commandes d'entraînement et ton modèle commencera à s'entraîner.

Fig 1. Nicolai Nielsen expliquant comment entraîner des modèles sur Ultralytics HUB.

Link to this sectionSuivi de la progression de l'entraînement#

Ultralytics HUB fournit un suivi en temps réel de la progression de l'entraînement de ton modèle. Tu peux surveiller des métriques clés telles que la précision, le rappel et les fonctions de perte. La plateforme visualise également les données d'entraînement, te permettant de voir comment ton modèle s'améliore au fil du temps.

Link to this sectionFonctionnalités avancées et intégrations#

Link to this sectionEntraînement dans le cloud et intégration Roboflow#

Ultralytics HUB a introduit l'entraînement dans le cloud, permettant aux utilisateurs d'entraîner des modèles directement dans le cloud sans aucune configuration locale. La plateforme s'intègre également avec Roboflow pour annoter des images et préparer des jeux de données. Cette intégration rationalise le flux de travail de la préparation des données à l'entraînement du modèle.

Link to this sectionApplication mobile#

L'application Ultralytics HUB apporte la détection d'objets en temps réel sur ton appareil mobile. En utilisant des modèles pré-entraînés, l'application peut détecter des objets du dataset COCO avec une grande précision, fonctionnant à 30 images par seconde sur un iPhone 14 Pro. Cette fonctionnalité est parfaite pour tester des modèles dans des scénarios réels et démontrer les capacités de tes modèles entraînés.

Link to this sectionConclusion#

Ultralytics HUB change la donne dans le monde de la vision par ordinateur, rendant plus facile que jamais l'entraînement de modèles YOLO personnalisés. Son interface conviviale, ses fonctionnalités robustes et ses intégrations transparentes en font un outil indispensable pour les développeurs et les chercheurs. Que tu cherches à entraîner des modèles localement, dans le cloud ou en déplacement avec l'application mobile, Ultralytics HUB est là pour toi.

Alors pourquoi attendre ? Plonge dans Ultralytics HUB dès aujourd'hui et libère le potentiel de tes projets de machine learning en quelques clics !