Ultralytics YOLO27:

Visualizza le architetture YOLO con YAML2ModelGraph#

YAML2ModelGraph è uno strumento della community che converte i file YAML dei modelli Ultralytics YOLO in diagrammi dell'architettura SVG. Può aiutarti a documentare modelli personalizzati o a esaminare la struttura del backbone, del neck e della head.

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

Installazione#

Clona il progetto e installa la sua dipendenza PyYAML:

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

Utilizzo#

Genera un diagramma dall'esempio YOLO26 incluso:

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

Usa --head multi per disegnare nodi head separati per scala oppure seleziona uno dei temi inclusi:

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

I temi disponibili includono paper, paper_ryb, candy, dark, ocean, retro, blueprint, forest e journal. L'SVG generato non richiede l'installazione di Graphviz.

Metadati dei nodi#

Modifica DISPLAY_CONFIG nel file main.py di YAML2ModelGraph per scegliere quali dettagli visualizzare in ogni nodo:

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

I nomi dei moduli YAML personalizzati vengono visualizzati automaticamente. I colori, le forme e la tipografia dei temi sono configurati nel themes.py dello strumento.

Limitazioni#

YAML2ModelGraph allinea in modo specifico le head standard Detect. Altre head delle attività, come OBB, Pose e Segment, vengono attualmente visualizzate come nodi generici del livello neck. Consulta il repository YAML2ModelGraph per informazioni aggiornate sul supporto e sull'utilizzo.

FAQ#

  • Sì. Qualsiasi file YAML di un modello Ultralytics che segue il formato dei layer [from, repeats, module, args] viene visualizzato e i nomi dei moduli personalizzati vengono disegnati come nodi generici. Consulta la model YAML configuration guide per il formato.

  • No. Lo strumento scrive direttamente in formato SVG e dipende esclusivamente da PyYAML.

  • Solo le teste standard Detect. Le teste OBB, Pose e Segment sono disegnate come nodi generici della corsia del collo; controlla il YAML2ModelGraph repository per il supporto attuale.

Contributori
GLglenn-jocher1RARahulbiradar91

Commenti