Visualizza le architetture YOLO con YAML2ModelGraph#

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

Installazione#

Clona il progetto e installa la 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 distinti per ogni scala oppure scegli 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. Per generare l'SVG non serve installare Graphviz.

Metadati dei nodi#

Modifica DISPLAY_CONFIG in main.py di YAML2ModelGraph per scegliere quali dettagli mostrare in ciascun 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 si configurano in themes.py dello strumento.

Limitazioni#

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

Domande frequenti#