Comment entraîner vos modèles personnalisés avec Ultralytics HUB

Nuvola Ladi

3 minutes de lecture

31 mai 2024

Découvrez comment former des modèles YOLO personnalisés sans effort avec Ultralytics HUB. Ce guide étape par étape de la plateforme intuitive offre un aperçu de l'intégration transparente, du suivi en temps réel et de l'entraînement dans le nuage.

Alors que nous nous efforçons constamment d'innover dans le domaine de l'IA, nous vous invitons à découvrir Ultralytics HUB, une plateforme conçue pour simplifier l'apprentissage des modèles YOLO d'Ultralytics. Que vous soyez un développeur chevronné ou un débutant, Ultralytics HUB offre une expérience transparente pour la création et la gestion de modèles de vision par ordinateur. Voyons comment entraîner vos modèles YOLO personnalisés en quelques clics.

Qu'est-ce que le HUB Ultralytics ?

Ultralytics HUB est une plateforme complète conçue pour gérer vos ensembles de données, vos projets et vos modèles, facilitant ainsi la formation et le déploiement de modèles d'apprentissage automatique. 

Il 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 en savoir plus sur les aspects techniques et le déroulement des opérations étape par étape, consultez notre article Medium sur la formation de modèles de vision par ordinateur dans le cloud.

Démarrer avec Ultralytics HUB

Interface conviviale

La première chose que vous remarquerez à propos d'Ultralytics HUB est son interface intuitive. La page d'accueil offre un accès facile aux ensembles de données, aux projets, aux modèles et aux intégrations. Bien qu'il s'agisse encore d'une version bêta, la plateforme dispose déjà d'une série de fonctionnalités puissantes.

Ensembles de données et modèles

Ultralytics HUB est livré préchargé avec des ensembles de données standard tels que VOC, COCO et Simpsons, qui sont couramment utilisés pour l'évaluation comparative. Ces ensembles de données sont importants, COCO comprenant 80 classes et 140 000 images. Cependant, la plateforme prend également en charge des ensembles de données plus petits, ce qui la rend idéale pour les utilisateurs de tous niveaux. Vous pouvez télécharger votre ensemble de données personnalisé ou vous connecter à des outils externes tels que Roboflow pour l'annotation des images et la préparation des ensembles de données.

Création d'un projet

La création d'un nouveau projet dans Ultralytics HUB est incroyablement simple. Voici un guide étape par étape pour vous aider à démarrer :

  1. Créer un projet : Commencez par donner un nom à votre projet. Pour cet exemple, appelons-le "Détection d'objets".
  2. Choose a Dataset (Choisir un ensemble de données) : Sélectionnez un ensemble de données parmi les options disponibles. Pour simplifier, nous utiliserons l'ensemble de données Simpsons avec 14 000 images.
  3. Sélectionnez un modèle : Choisissez un modèle YOLO. Ultralytics HUB prend en charge différents modèles, dont YOLOv5 et YOLOv8. Pour les besoins de cet exemple, nous choisirons le modèle YOLOv8 nano.
  4. Ajuster les hyperparamètres : Ajustez les paramètres tels que le nombre d'époques, la taille de l'image et la taille du lot. Vous pouvez également choisir d'utiliser un GPU ou un CPU et configurer les options de mise en cache.

Formation du modèle

Une fois votre projet mis en place, il suffit d'un clic pour former le modèle. Ultralytics HUB offre plusieurs options de formation :

  • Formation locale : Installez Ultralytics sur votre machine locale et exécutez le script de formation.
  • Google Colab: Ouvrez un carnet Google Colab préconfiguré avec le code nécessaire. Cette option ne nécessite aucun codage et fonctionne de manière transparente dans le nuage.
  • HUB Cloud : La formation Ultralytics HUB Cloud offre une solution sans code pour l'entraînement des modèles YOLO, parfaite pour les non-codeurs et les propriétaires d'entreprise. Le flux de travail comprend le téléchargement des données, la sélection du modèle et la configuration de l'instance cloud, ce qui permet d'affiner les modèles pré-entraînés et de les exporter pour diverses applications.

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

  1. Installez Ultralytics : Exécutez la commande pour installer Ultralytics dans le carnet.
  2. Configuration et clé API : Configurez l'installation et entrez votre clé API.
  3. Commencer la formation : Exécutez les commandes de formation et votre modèle commencera la formation.
__wf_reserved_inherit
Fig 1. Nicolai Nielsen explique comment former des modèles sur Ultralytics HUB.

Suivi des progrès de la formation

Ultralytics HUB permet de suivre en temps réel la progression de l'entraînement de votre modèle. Vous pouvez surveiller des paramètres clés tels que l'exactitude, la précision, le rappel et les fonctions de perte. La plateforme visualise également les données d'entraînement, ce qui vous permet de voir comment votre modèle s'améliore au fil du temps.

Fonctionnalités avancées et intégrations

Formation à l'informatique en nuage et intégration de RoboFlow

Ultralytics HUB a introduit la formation en nuage, permettant aux utilisateurs de former des modèles directement dans le nuage sans aucune installation locale. La plateforme s'intègre également à Roboflow pour l'annotation des images et la préparation des ensembles de données. Cette intégration rationalise le flux de travail, de la préparation des données à l'entraînement des modèles.

Application mobile

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

Conclusion

Ultralytics HUB change la donne dans le monde de la vision par ordinateur, en facilitant plus 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 vous cherchiez à entraîner des modèles localement, dans le nuage ou en déplacement avec l'application mobile, Ultralytics HUB vous couvre.

Alors pourquoi attendre ? Plongez dans Ultralytics HUB dès aujourd'hui et libérez le potentiel de vos projets d'apprentissage automatique en quelques clics !

Construisons ensemble le futur
de l'IA !

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

Commencer gratuitement
Lien copié dans le presse-papiers