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

Комментарии