Ultralytics YOLO27:

Visualiza arquitecturas de YOLO con YAML2ModelGraph#

YAML2ModelGraph es una herramienta comunitaria que convierte archivos YAML de modelos de Ultralytics YOLO en diagramas de arquitectura SVG. Puede ayudarte a documentar modelos personalizados o inspeccionar la estructura de su backbone, neck y head.

Single-head YOLO architecture rendered by YAML2ModelGraph Multi-head YOLO architecture rendered by YAML2ModelGraph

Instalación#

Clona el proyecto e instala su dependencia PyYAML:

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

Uso#

Genera un diagrama a partir del ejemplo de YOLO26 incluido:

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

Usa --head multi para dibujar nodos head independientes específicos de cada escala, o selecciona uno de los temas incluidos:

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

Entre los temas disponibles se incluyen paper, paper_ryb, candy, dark, ocean, retro, blueprint, forest y journal. El SVG generado no requiere instalar Graphviz.

Metadatos de los nodos#

Edita DISPLAY_CONFIG en main.py de YAML2ModelGraph para elegir qué detalles aparecen en cada nodo:

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

Los nombres de módulos YAML personalizados se representan automáticamente. Los colores, las formas y la tipografía del tema se configuran en el themes.py de la herramienta.

Limitaciones#

YAML2ModelGraph alinea especialmente los heads estándar de Detect. Otros heads de tareas, como OBB, Pose y Segment, se representan actualmente como nodos genéricos del carril del neck. Consulta el repositorio de YAML2ModelGraph para obtener información actualizada sobre la compatibilidad y el uso.

Preguntas frecuentes#

  • Sí. Cualquier archivo YAML de modelos de Ultralytics que siga el formato de capa [from, repeats, module, args] se renderiza, y los nombres de los módulos personalizados se dibujan como nodos genéricos. Consulta la model YAML configuration guide para conocer el formato.

  • No. La herramienta genera SVG directamente y solo depende de PyYAML.

  • Solo los cabezales estándar de Detect. Los cabezales OBB, Pose y Segment se dibujan como nodos genéricos de la ruta del cuello; consulta el YAML2ModelGraph repository para ver la compatibilidad actual.

Colaboradores
GLglenn-jocher1RARahulbiradar91

Comentarios