Visualize arquiteturas YOLO com YAML2ModelGraph#

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

Instalação#

Clone o projeto e instale a dependência PyYAML:

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

Uso#

Gere um diagrama usando o exemplo YOLO26 incluído:

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

Use --head multi para desenhar nós head separados para cada escala ou selecione 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#

Edite DISPLAY_CONFIG em main.py do YAML2ModelGraph para escolher quais 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 renderizados automaticamente. As cores do tema, as formas e a tipografia são configuradas em themes.py da ferramenta.

Limitações#

O YAML2ModelGraph alinha especialmente os heads padrão Detect. Outros heads de tarefa, como Segment, Pose e OBB, são atualmente renderizados como nós genéricos na faixa do neck. Consulte o repositório YAML2ModelGraph para ver os detalhes atuais sobre suporte e uso.

Perguntas frequentes#