YOLOv8 vs YOLOv5#

Choisir la bonne architecture de vision par ordinateur est une étape critique dans la construction de pipelines d'apprentissage automatique robustes. Dans cette comparaison technique détaillée, nous explorons les différences entre deux des modèles les plus populaires de l'écosystème de l'IA visuelle : YOLOv8 et YOLOv5. Les deux modèles ont été développés par Ultralytics et ont façonné de manière significative le paysage de la détection d'objets en temps réel, établissant des normes industrielles en matière de vitesse, de précision et de facilité d'utilisation.

Que tu déploies sur des appareils de périphérie ou que tu mettes à l'échelle l'inférence dans le cloud, comprendre les évolutions architecturales, les métriques de performance et les méthodologies d'entraînement de ces modèles t'aidera à prendre une décision éclairée pour tes projets de vision par ordinateur.

Ultralytics YOLOv8 : le standard polyvalent#

Sorti au début de l'année 2023, YOLOv8 a représenté un changement architectural majeur par rapport à ses prédécesseurs. Il a été conçu dès le départ pour servir de cadre unifié capable de gérer nativement plusieurs tâches de vision, notamment la segmentation d'instances, la classification d'images et l'estimation de pose.

• Auteurs : Glenn Jocher, Ayush Chaurasia et Jing Qiu
• Organisation : Ultralytics
• Date : 2023-01-10
• GitHub : ultralytics/ultralytics
• Docs : Documentation YOLOv8

Architecture et méthodologies#

YOLOv8 a introduit une tête de détection sans ancrage (anchor-free), ce qui simplifie le processus d'entraînement en éliminant le besoin de configurer manuellement des boîtes d'ancrage en fonction de la distribution du jeu de données. Cela rend le modèle plus robuste lors de la généralisation à des jeux de données personnalisés et réduit le nombre de prédictions de boîtes, accélérant ainsi la suppression des non-maximaux (NMS).

L'architecture comporte un module C2f (Cross-Stage Partial bottleneck avec deux convolutions), qui remplace le module C3 présent dans YOLOv5. Le module C2f améliore le flux de gradient et permet au modèle d'apprendre des représentations de caractéristiques plus riches sans augmentation significative du coût computationnel. De plus, YOLOv8 utilise une structure de tête découplée, séparant les tâches d'objectness, de classification et de régression, ce qui a démontré une amélioration de la vitesse de convergence et de la précision.

Points forts et faiblesses#

Points forts :

• Polyvalence inégalée sur plusieurs tâches au-delà de la simple détection par boîte englobante.
• API Python rationalisée via le paquet ultralytics, rendant l'entraînement et l'exportation extrêmement intuitifs.
• Précision moyenne (mAP) mean Average Precision (mAP) plus élevée sur toutes les variantes de taille par rapport à YOLOv5.

Points faibles :

• La tête découplée et le module C2f introduisent une légère augmentation du nombre de paramètres et de FLOPs pour certaines variantes par rapport à leurs homologues exacts de YOLOv5.

En savoir plus sur YOLOv8

Ultralytics YOLOv5 : Le pionnier agile#

Introduit en 2020, YOLOv5 a amené YOLO dans l'écosystème PyTorch, améliorant considérablement l'accessibilité pour les développeurs. Il est rapidement devenu la norme industrielle pour les modèles de détection d'objets rapides, fiables et faciles à déployer.

• Auteur : Glenn Jocher
• Organisation : Ultralytics
• Date : 2020-06-26
• GitHub : ultralytics/yolov5
• Documentation : YOLOv5 Documentation

Architecture et méthodologies#

YOLOv5 repose sur une architecture basée sur des ancres et utilise une dorsale (backbone) CSPDarknet53 modifiée. Bien que les approches basées sur des ancres nécessitent un regroupement minutieux des boîtes englobantes du jeu de données pour définir des ancres optimales avant l'entraînement, elles sont très efficaces pour des jeux de données spécifiques et bien définis.

YOLOv5 intègre le module C3, qui extrait efficacement les caractéristiques tout en maintenant une faible empreinte en termes de paramètres. Sa fonction de perte repose largement sur la perte d'Objectness combinée aux pertes de classification et de régression des boîtes englobantes pour guider le réseau vers des prédictions précises.

Points forts et faiblesses#

Points forts :

• Extrêmement léger, ce qui rend les variantes Nano (YOLOv5n) et Small (YOLOv5s) particulièrement adaptées aux déploiements d'IA en périphérie aux ressources contraintes.
• Vitesses d'inférence exceptionnellement rapides, surtout sur CPU.
• Un écosystème profondément établi avec de vastes tutoriels communautaires et des intégrations tierces.

Points faibles :

• Nécessite une configuration des boîtes d'ancrage, ce qui peut compliquer la mise en place pour des jeux de données très variés ou personnalisés.
• Précision globale (mAP) inférieure par rapport aux architectures modernes sans ancres comme YOLOv8 et YOLO26.

En savoir plus sur YOLOv5

Comparaison des performances#

Lors de l'évaluation de ces modèles, l'obtention d'un compromis favorable entre vitesse et précision est primordiale. Le tableau ci-dessous présente les métriques de performance des deux architectures évaluées sur le jeu de données COCO. Les vitesses sur CPU ont été mesurées à l'aide d'ONNX, tandis que les vitesses sur GPU ont été testées à l'aide de TensorRT.

Alors que YOLOv5 conserve un léger avantage en nombre de paramètres et en vitesse brute absolue pour sa variante Nano, YOLOv8 offre un saut massif en mAP sur toute la ligne, offrant un meilleur équilibre de performance pour des scénarios de déploiement réel exigeants.

Facilité d'utilisation et l'écosystème Ultralytics#

Une caractéristique déterminante des modèles Ultralytics modernes est l'écosystème bien entretenu qui les entoure. La transition de YOLOv5 à YOLOv8 a introduit le paquet pip unifié ultralytics, créant une expérience utilisateur extrêmement rationalisée.

Les développeurs peuvent gérer de manière transparente l'entraînement de modèles, la validation, la prédiction et l'exportation avec seulement quelques lignes de code Python, en contournant les scripts de code passe-partout complexes historiquement requis dans les projets de deep learning.

from ultralytics import YOLO

# Load a pretrained YOLOv8 model
model = YOLO("yolov8n.pt")

# Train the model on custom data efficiently
results = model.train(data="coco8.yaml", epochs=100, imgsz=640, device=0)

# Export the model to ONNX for production deployment
path = model.export(format="onnx")

De plus, l'intégration avec des outils tels que Ultralytics Platform simplifie la gestion des jeux de données, l'entraînement dans le cloud et le déploiement, garantissant un développement actif et un fort support communautaire.

Cas d'utilisation idéaux#

Quand choisir YOLOv5 : Si tu maintiens des systèmes existants, que tu exécutes des inférences sur des CPU sévèrement contraints comme un Raspberry Pi, ou que tu travailles sur un projet où économiser chaque fraction de mégaoctet dans la taille du modèle est critique, YOLOv5 reste un outil de travail fiable.

Quand choisir YOLOv8 : Pour pratiquement tous les nouveaux projets commençant aujourd'hui, YOLOv8 est fortement recommandé par rapport à YOLOv5. Son architecture avancée gère le suivi complexe, les boîtes englobantes orientées (OBB) et la segmentation sans effort. Il est idéal pour les applications modernes allant de la robotique autonome à l'analyse d'images médicales et aux infrastructures de villes intelligentes.

En tirant parti de la documentation complète et des outils fournis par Ultralytics, tu peux facilement déployer YOLOv8, ou explorer le cutting-edge YOLO26, pour résoudre des défis visuels complexes avec une vitesse et une précision sans précédent. Pour aller plus loin, pense à explorer nos guides sur le réglage des hyperparamètres et les pratiques de déploiement de modèles.