YOLO Vision 2026:

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

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

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

Установка#

Клонируй проект и установи его зависимость 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 для отрисовки отдельных узлов голов для конкретных масштабов или выбери одну из доступных тем:

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 особым образом выравнивает стандартные головы Detect. Другие головы задач, такие как OBB, Pose и Segment, в настоящее время отображаются как общие узлы дорожки шеи. Подробную информацию о текущей поддержке и использовании см. в репозитории YAML2ModelGraph.

Участники
RARahulbiradar91

Комментарии