Ultralytics YOLO27 :
Get Started

Visualiser les architectures YOLO avec YAML2ModelGraph#

YAML2ModelGraph est un outil communautaire qui convertit les fichiers YAML de modèles Ultralytics YOLO en diagrammes d'architecture SVG. Il peut aider à documenter des modèles personnalisés ou à examiner la structure de leur backbone, neck et head.

Single-head YOLO architecture rendered by YAML2ModelGraph Multi-head YOLO architecture rendered by YAML2ModelGraph

Installation#

Clone le projet et installe sa dépendance PyYAML :

git clone https://github.com/WangQvQ/YAML2ModelGraph.git
cd YAML2ModelGraph
pip install pyyaml

Utilisation#

Génère un diagramme à partir de l'exemple YOLO26 inclus :

python main.py examples/yolo26.yaml output.svg

Utilise --head multi pour dessiner des nœuds de tête distincts pour chaque échelle, ou sélectionne l'un des thèmes inclus :

python main.py examples/yolo26.yaml output.svg --head multi --theme paper_ryb

Les thèmes disponibles incluent paper, paper_ryb, candy, dark, ocean, retro, blueprint, forest et journal. Le SVG généré ne nécessite aucune installation de Graphviz.

Métadonnées des nœuds#

Modifie DISPLAY_CONFIG dans main.py de YAML2ModelGraph pour choisir les détails qui apparaissent dans chaque nœud :

DISPLAY_CONFIG = {
    "show_channels": True,
    "show_repeats": True,
    "show_stride": True,
    "show_args": False,
}

Les noms de modules YAML personnalisés sont rendus automatiquement. Les couleurs, les formes et la typographie des thèmes se configurent dans le themes.py de l'outil.

Limites#

YAML2ModelGraph aligne spécifiquement les têtes Detect standard. Les autres têtes de tâche, comme Segment, Pose et OBB, sont actuellement représentées par des nœuds génériques dans la voie du neck. Consulte le dépôt YAML2ModelGraph pour connaître la prise en charge et les modalités d'utilisation actuelles.

FAQ#

  • Oui. Tout fichier YAML de modèle Ultralytics respectant le format de couche [from, repeats, module, args] est rendu, et les noms de modules personnalisés sont représentés par des nœuds génériques. Consulte le guide de configuration YAML des modèles pour connaître le format.

  • Non. L'outil écrit directement le SVG et dépend uniquement de PyYAML.

  • Uniquement les têtes Detect standard. Les têtes Segment, Pose et OBB sont représentées par des nœuds génériques dans la voie du neck ; consulte le dépôt YAML2ModelGraph pour connaître la prise en charge actuelle.

Contributeurs
GLglenn-jocher2RARahulbiradar91

Commentaires