使用 YAML2ModelGraph 可视化 YOLO 架构#

YAML2ModelGraph 是一个社区工具，可将 Ultralytics YOLO 模型 YAML 文件转换为 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 绘制独立的特定规模头部节点，或选择包含的主题之一：

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。

节点元数据#

在 YAML2ModelGraph 的 main.py 中编辑 DISPLAY_CONFIG，以选择每个节点中显示的详细信息：

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

自定义 YAML 模块名称会自动渲染。主题颜色、形状和排版在该工具的 themes.py 中进行配置。

局限性#

YAML2ModelGraph 对标准 Detect 头部进行了特殊对齐。其他任务头部（如 OBB、Pose 和 Segment）目前被渲染为通用的颈部通道节点。有关当前支持和使用详情，请参阅 YAML2ModelGraph 仓库。