Visualizza le architetture YOLO con YAML2ModelGraph#

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

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 della head specifici per scala separati, 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. Il file SVG generato non richiede alcuna installazione di Graphviz.

Metadati del nodo#

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

Limitazioni#

YAML2ModelGraph allinea in modo speciale le head Detect standard. Altre head di task come OBB, Pose e Segment sono attualmente visualizzate come nodi generici nella corsia del neck. Consulta il repository di YAML2ModelGraph per i dettagli aggiornati su supporto e utilizzo.