Visualize arquiteturas YOLO com o 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 de backbone, neck e head.

Instalação#

Clone o projeto e instale sua dependência PyYAML:

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

Uso#

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

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

Use --head multi para desenhar nós de head específicos de escala separados, 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 instalação do Graphviz.

Metadados do Nó#

Edite DISPLAY_CONFIG no 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,
}

Nomes de módulos YAML personalizados são renderizados automaticamente. As cores, formas e tipografia dos temas são configuradas no themes.py da ferramenta.

Limitações#

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