Визуализация архитектур YOLO с помощью YAML2ModelGraph#

YAML2ModelGraph — это инструмент сообщества, который преобразует YAML-файлы моделей Ultralytics YOLO в SVG-диаграммы архитектуры. Он помогает документировать пользовательские модели или изучать структуру их backbone, neck и head.

Установка#

Клонируй проект и установи его зависимость PyYAML:

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

Использование#

Сгенерируй диаграмму на основе включённого примера YOLO26:

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

Используй --head multi, чтобы отображать отдельные узлы head для каждого масштаба, или выбери одну из включённых тем:

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

Доступны темы paper, paper_ryb, candy, dark, ocean, retro, blueprint, forest и journal. Для сгенерированного SVG не требуется установка Graphviz.

Метаданные узлов#

Измени DISPLAY_CONFIG в main.py YAML2ModelGraph, чтобы выбрать сведения, отображаемые в каждом узле:

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

Имена пользовательских модулей YAML отображаются автоматически. Цвета тем, формы и типографика настраиваются в themes.py этого инструмента.

Ограничения#

YAML2ModelGraph специально выравнивает стандартные узлы head Detect. Другие узлы head для задач, такие как OBB, Pose и Segment, в настоящее время отображаются как универсальные узлы в полосе neck. Актуальные сведения о поддержке и использовании см. в репозитории YAML2ModelGraph.

Часто задаваемые вопросы#