Passer au contenu

YOLOv9 vs. DAMO-YOLO : une comparaison technique complète

Dans le paysage en évolution rapide de la vision par ordinateur, le choix de l'architecture de détection d'objets optimale est essentiel pour la réussite d'un projet. Cette analyse fournit une comparaison technique détaillée entre deux modèles formidables : YOLOv9, salué pour ses innovations architecturales en matière d'informations de gradient, et DAMO-YOLO, un modèle d'Alibaba Group conçu pour l'inférence à haute vitesse. Nous examinons leurs architectures uniques, leurs mesures de performance et leurs scénarios de déploiement idéaux pour guider les développeurs et les chercheurs dans la prise de décisions éclairées.

YOLOv9 : Informations de gradient programmables pour une précision supérieure

YOLOv9 marque une évolution significative dans la série You Only Look Once (YOLO), en se concentrant sur la résolution du problème du goulot d'étranglement de l'information inhérent aux réseaux neuronaux profonds. En veillant à ce que les données d'entrée cruciales soient préservées tout au long des couches du réseau, YOLOv9 atteint une précision de pointe.

Auteurs: Chien-Yao Wang et Hong-Yuan Mark Liao
Organisation:Institute of Information Science, Academia Sinica, Taiwan
Date: 2024-02-21
Arxiv:2402.13616
GitHub:WongKinYiu/yolov9
Docs:Documentation Ultralytics YOLOv9

Architecture et innovations de base

L'architecture de YOLOv9 est basée sur deux concepts révolutionnaires conçus pour optimiser l'efficacité de l'apprentissage profond :

  1. Information de Gradient Programmable (PGI) : PGI est un cadre de supervision auxiliaire qui traite le problème de la perte d'informations lorsque les données se propagent à travers les couches profondes. Il garantit que la fonction de perte reçoit des gradients fiables, permettant au modèle d'apprendre des caractéristiques plus efficaces sans augmenter le coût d'inférence.
  2. Generalized Efficient Layer Aggregation Network (GELAN) : Cette nouvelle architecture combine les forces de CSPNet et d’ELAN. GELAN est conçu pour maximiser l’utilisation des paramètres et l’efficacité de calcul, fournissant un backbone léger mais puissant qui prend en charge divers blocs de calcul.

Forces et écosystème

  • Précision de premier ordre : YOLOv9 atteint des scores mAP exceptionnels sur l'ensemble de données COCO, établissant ainsi des références pour les détecteurs d’objets en temps réel.
  • Efficacité des paramètres : Grâce à GELAN, le modèle offre de hautes performances avec moins de paramètres par rapport à de nombreux prédécesseurs.
  • Intégration Ultralytics : Faire partie de l’écosystème Ultralytics signifie que YOLOv9 bénéficie d’une API Python unifiée, d’options d’exportation de modèles fluides (ONNX, TensorRT, CoreML) et d’une documentation robuste.
  • Stabilité de l'entraînement : Le framework PGI améliore considérablement la vitesse de convergence et la stabilité pendant l'entraînement du modèle.

Faiblesses

  • Intensité des ressources : Bien qu'elle soit efficace pour sa classe de précision, les variantes les plus grandes (comme YOLOv9-E) nécessitent une mémoire GPU importante pour l'entraînement.
  • Priorité de la tâche : La recherche principale cible principalement la détection d’objets, tandis que d’autres modèles Ultralytics comme YOLO11 prennent en charge nativement un éventail plus large de tâches, y compris l’estimation de la pose et l’OBB, dès le départ.

En savoir plus sur YOLOv9

DAMO-YOLO : Recherche d'architecture neuronale pour la vitesse

DAMO-YOLO témoigne de la puissance de la conception d'architecture automatisée. Développé par Alibaba, il exploite la recherche d'architecture neuronale (NAS) pour trouver l'équilibre optimal entre la latence d'inférence et les performances de détection, ciblant spécifiquement les applications industrielles.

Auteurs : Xianzhe Xu, Yiqi Jiang, Weihua Chen, Yilun Huang, Yuan Zhang et Xiuyu Sun
Organisation :Alibaba Group
Date : 2022-11-23
Arxiv :2211.15444
GitHub :tinyvision/DAMO-YOLO

Architecture et principales fonctionnalités

DAMO-YOLO se distingue par plusieurs avancées technologiques visant à maximiser le débit :

  • Réseau MAE-NAS : Il utilise une structure dorsale dérivée de la méthode Method-Aware Efficient Neural Architecture Search, optimisant la topologie du réseau pour des contraintes matérielles spécifiques.
  • RepGFPN efficace : Le modèle emploie un réseau de pyramide de caractéristiques généralisé reparamétré pour son neck, améliorant la fusion des caractéristiques tout en maintenant une faible latence.
  • ZeroHead : Une conception de tête de détection légère qui réduit la surcharge de calcul généralement associée aux couches de prédiction finales.
  • AlignedOTA : Une stratégie améliorée d’attribution d’étiquettes qui résout le problème de désalignement entre les tâches de classification et de régression pendant la formation.

Points forts

  • Faible latence : DAMO-YOLO est conçu pour la vitesse, ce qui le rend très efficace pour l'inférence en temps réel sur les appareils périphériques et les GPU.
  • Conception automatisée : L'utilisation de NAS garantit que l'architecture est mathématiquement optimisée pour l'efficacité plutôt que de s'appuyer uniquement sur des heuristiques manuelles.
  • Sans ancres : Il adopte une approche sans ancres, simplifiant ainsi le processus de réglage des hyperparamètres liés aux boîtes d'ancrage.

Faiblesses

  • Écosystème limité : Comparé à l'outillage étendu disponible pour les modèles Ultralytics, DAMO-YOLO a une communauté plus petite et moins d'outils d'intégration prêts à l'emploi pour le MLOps.
  • Polyvalence : Il est principalement spécialisé dans la détection, sans les capacités natives multi-tâches (segmentation, classification) que l’on trouve dans les frameworks plus complets.

En savoir plus sur DAMO-YOLO

Analyse des performances : Vitesse vs. Précision

Lorsque l'on compare les mesures de performance, les compromis entre les deux architectures deviennent clairs. YOLOv9 donne la priorité à la préservation de l'information pour atteindre une précision supérieure, dépassant souvent DAMO-YOLO dans les scores mAP pour des tailles de modèle similaires. Inversement, DAMO-YOLO se concentre sur le débit brut.

Cependant, l'efficacité de l'architecture GELAN de YOLOv9 lui permet de rester très compétitif en termes de vitesse tout en offrant une meilleure qualité de détection. Par exemple, YOLOv9-C atteint un mAP nettement supérieur (53,0 %) par rapport à DAMO-YOLO-L (50,8 %) tout en utilisant moins de paramètres (25,3M contre 42,1M). Cela souligne la capacité de YOLOv9 à offrir « plus pour moins » en termes de complexité du modèle.

Interprétation des performances

Lors de l'évaluation des modèles, tenez compte des FLOPs (opérations en virgule flottante) ainsi que du nombre de paramètres. Un nombre de FLOPs plus faible indique généralement un modèle qui est plus léger en termes de calcul et potentiellement plus rapide sur le matériel mobile ou edge AI.

ModèleTaille
(pixels)
mAPval
50-95
Vitesse
CPU ONNX
(ms)
Vitesse
T4 TensorRT10
(ms)
paramètres
(M)
FLOPs
(B)
YOLOv9t64038.3-2.32.07.7
YOLOv9s64046.8-3.547.126.4
YOLOv9m64051.4-6.4320.076.3
YOLOv9c64053.0-7.1625.3102.1
YOLOv9e64055.6-16.7757.3189.0
DAMO-YOLOt64042.0-2.328.518.1
DAMO-YOLOs64046.0-3.4516.337.8
DAMO-YOLOm64049.2-5.0928.261.8
DAMO-YOLOl64050.8-7.1842.197.3

Cas d'utilisation idéaux

Les différences architecturales dictent les scénarios de déploiement idéaux pour chaque modèle.

Applications de YOLOv9

YOLOv9 est le choix préféré pour les applications où la précision est non négociable.

  • Imagerie médicale : Détection d’anomalies subtiles dans l'analyse d’images médicales où le fait de manquer une détection pourrait être critique.
  • Navigation autonome : Systèmes de perception avancés pour les voitures autonomes nécessitant une grande confiance dans la détection d'objets.
  • Surveillance détaillée : Systèmes de sécurité qui doivent identifier de petits objets ou fonctionner dans des environnements complexes avec un encombrement élevé.

Applications de DAMO-YOLO

DAMO-YOLO excelle dans les environnements contraints par des budgets de latence stricts.

  • Fabrication à haute vitesse : Lignes industrielles où les systèmes de vision par ordinateur doivent suivre le rythme des chaînes de montage rapides.
  • Analyse vidéo : Traitement de volumes massifs de flux vidéo où le coût de débit est une préoccupation majeure.

L'avantage Ultralytics

Bien que les deux modèles soient techniquement impressionnants, le choix d'un modèle au sein de l'écosystème Ultralytics—tel que YOLOv9 ou le YOLO11 de pointe YOLO11—offre des avantages distincts pour les développeurs et les entreprises.

Flux de travail et convivialité simplifiés

Ultralytics privilégie la facilité d'utilisation. Les modèles sont accessibles via une interface unifiée qui fait abstraction du code standard complexe. Que vous vous entraîniez sur des données personnalisées ou que vous fassiez de l'inférence, le processus est cohérent et intuitif.

from ultralytics import YOLO

# Load a pre-trained YOLOv9 model
model = YOLO("yolov9c.pt")

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

# Run inference on an image
results = model("path/to/image.jpg")

Écosystème bien entretenu

Les modèles Ultralytics sont soutenus par une communauté active et des mises à jour fréquentes. Des fonctionnalités telles que Ultralytics HUB permettent la gestion des ensembles de données et la formation sur le Web, tandis que les intégrations étendues avec des outils tels que TensorBoard et MLflow rationalisent le cycle de vie des MLOps. En revanche, les modèles de recherche tels que YOLO manquent souvent de ce niveau de support continu et d'intégration d'outils.

Polyvalence et efficacité

Les modèles Ultralytics sont conçus pour être polyvalents. Alors que DAMO-YOLO est spécifique à la détection, les modèles Ultralytics comme YOLO11 étendent les capacités à la segmentation d'instance, à l'estimation de pose et à la détection de boîtes englobantes orientées (OBB). De plus, ils sont optimisés pour l'efficacité de la mémoire, nécessitant souvent moins de mémoire CUDA pendant l'entraînement par rapport à d'autres architectures, ce qui permet de réduire les coûts matériels.

Conclusion

Dans la comparaison de YOLOv9 vs. DAMO-YOLO, les deux modèles mettent en évidence les progrès rapides de l'IA. DAMO-YOLO offre une architecture convaincante pour l'optimisation de la vitesse pure. Cependant, YOLOv9 se distingue comme la solution la plus robuste pour la plupart des applications pratiques. Il offre une précision supérieure par paramètre, utilise une architecture avancée pour éviter la perte d'informations et réside dans l'écosystème Ultralytics florissant. Pour les développeurs à la recherche du meilleur équilibre entre performances, facilité d'utilisation et support à long terme, les modèles Ultralytics restent le choix recommandé.

Explorer d'autres modèles

Découvrez comment d'autres modèles de pointe se comparent dans notre documentation :


Commentaires