YOLO Vision 2026:

Visualizza le architetture YOLO con YAML2ModelGraph#

YAML2ModelGraph è uno strumento della community che converte i file YAML dei modelli di Ultralytics YOLO in diagrammi di architettura SVG. Può aiutarti a documentare modelli personalizzati o a ispezionare la loro struttura di backbone, neck e 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 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. L'SVG generato non richiede alcuna installazione di Graphviz.

Metadati dei nodi#

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

Limitazioni#

YAML2ModelGraph allinea in modo speciale gli head Detect standard. Altri head di task come OBB, Pose e Segment vengono attualmente visualizzati come nodi di corsia neck generici. Consulta il repository di YAML2ModelGraph per i dettagli attuali sul supporto e sull'utilizzo.

Collaboratori
RARahulbiradar91

Commenti