Visualiza arquiteturas YOLO com YAML2ModelGraph#

YAML2ModelGraph é uma ferramenta da comunidade que converte ficheiros YAML de modelos Ultralytics YOLO em diagramas de arquitetura SVG. Pode ajudar a documentar modelos personalizados ou a inspecionar a estrutura do backbone, neck e head.

Instalação#

Clona o projeto e instala a dependência PyYAML:

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

Utilização#

Gera um diagrama a partir do exemplo YOLO26 incluído:

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

Usa --head multi para desenhar nós de head separados por escala ou seleciona um dos temas incluídos:

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

Os temas disponíveis incluem paper, paper_ryb, candy, dark, ocean, retro, blueprint, forest e journal. O SVG gerado não requer a instalação do Graphviz.

Metadados dos nós#

Edita DISPLAY_CONFIG no main.py do YAML2ModelGraph para escolher que detalhes aparecem em cada nó:

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

Os nomes de módulos YAML personalizados são apresentados automaticamente. As cores, formas e a tipografia do tema são configuradas no themes.py da ferramenta.

Limitações#

O YAML2ModelGraph alinha especialmente os heads padrão Detect. Outros heads de tarefas, como OBB, Pose e Segment, são atualmente apresentados como nós genéricos na faixa do neck. Consulta o repositório do YAML2ModelGraph para obter detalhes sobre o suporte e a utilização atuais.

Perguntas frequentes#