Comment former tes modèles personnalisés avec Ultralytics HUB

Alors que nous nous efforçons constamment d'innover grâce à l'IA, rejoins-nous pour examiner de plus près Ultralytics HUB, une plateforme conçue pour simplifier la formation de modèles personnalisés.Ultralytics YOLO modèles. Que tu sois 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. Explorons comment former tes modèles personnalisés YOLO en quelques clics.

Qu'est-ce que Ultralytics HUB ?

Ultralytics HUB est une plateforme complète conçue pour gérer tes ensembles de données, tes projets et tes modèles, ce qui facilite 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 à la fois sur Android et iOS), et évolue en permanence avec de nouvelles fonctionnalités. Pour plus d'aperçus techniques et un flux de travail étape par étape, consulte notre article Medium sur la façon de former des modèles de vision par ordinateur sur le cloud.

Démarrage avec Ultralytics HUB

Interface conviviale

La première chose que tu remarqueras à propos de Ultralytics HUB est son interface intuitive. La page d'accueil permet d'accéder facilement aux ensembles de données, aux projets, aux modèles et aux intégrations. Même si elle est encore en version bêta, la plateforme se targue déjà d'une série de fonctionnalités puissantes.

Jeux 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'analyse comparative. Ces ensembles de données sont importants, COCO comportant 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. Tu peux télécharger ton jeu de données personnalisé ou te 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 te permettre de commencer :

1. Crée un projet : Commence par donner un nom à ton projet. Pour cet exemple, appelons-le "Détection d'objets".
2. Choisir un jeu de données : Sélectionne un ensemble de données parmi les options disponibles. Pour plus de simplicité, nous utiliserons l'ensemble de données Simpsons avec 14 000 images.
3. Choisis un modèle : Choisis un modèle YOLO . Ultralytics HUB prend en charge différents modèles, notamment YOLOv5 et YOLOv8. Pour les besoins de cet exemple, nous choisirons le modèle YOLOv8 nano.
4. Régler les hyperparamètres : Ajuste les paramètres tels que le nombre d'époques, la taille de l'image et la taille du lot. Tu peux également choisir d'utiliser GPU ou CPU et configurer les options de mise en cache.

Former le modèle

Une fois ton projet mis en place, la formation au modèle se fait en un clic. Ultralytics HUB propose plusieurs options de formation :

• Formation locale : Installe Ultralytics sur ta machine locale et exécute le script de formation.
• Google Colab: Ouvre un carnet de notes Google Colab préconfiguré avec le code nécessaire. Cette option ne nécessite aucun codage et fonctionne de manière transparente dans le cloud.
• HUB Cloud : Le Ultralytics HUB Cloud Training offre une solution sans code pour la formation 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 jeux de données, la sélection des modèles et la configuration de l'instance cloud, ce qui permet d'affiner facilement les modèles pré-entraînés et de les exporter pour diverses applications.

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

1. Installer Ultralytics: Exécute la commande pour installer Ultralytics dans l'ordinateur portable.
2. Configuration et clé API : Configure l'installation et saisis ta clé API.
3. Commence la formation : Exécute les commandes de formation, et ton modèle commencera à s'entraîner.

Suivi des progrès de la formation

Ultralytics HUB permet de suivre en temps réel la progression de l'entraînement de ton modèle. Tu peux 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 te permet de voir comment ton modèle s'améliore au fil du temps.

Fonctions avancées et intégrations

Formation dans le nuage et intégration de RoboFlow

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

Application mobile

L'application Ultralytics HUB te permet de détecter des 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 à partir de l'ensemble de données COCO avec une grande précision, en 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 tes modèles entraînés.

Pour conclure

Ultralytics HUB change la donne dans le monde de la vision par ordinateur, en facilitant plus que jamais la formation de modèles personnalisés YOLO . Son interface conviviale, ses fonctionnalités robustes et ses intégrations transparentes le positionnent comme un outil incontournable pour les développeurs et les chercheurs. Que tu cherches à former des modèles localement, dans le nuage ou en déplacement avec l'application mobile, Ultralytics HUB te couvre.

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