Trực quan hóa Kiến trúc YOLO với YAML2ModelGraph#

YAML2ModelGraph là một công cụ cộng đồng chuyển đổi các tệp YAML của mô hình Ultralytics YOLO thành sơ đồ kiến trúc SVG. Công cụ này giúp tài liệu hóa các mô hình tùy chỉnh hoặc kiểm tra cấu trúc backbone, neck và head của chúng.

Cài đặt#

Sao chép dự án và cài đặt phụ thuộc PyYAML của dự án:

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

Cách sử dụng#

Tạo sơ đồ từ ví dụ YOLO26 đi kèm:

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

Sử dụng --head multi để vẽ các node head theo từng tỷ lệ riêng biệt, hoặc chọn một trong các chủ đề đi kèm:

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

Các chủ đề có sẵn bao gồm paper, paper_ryb, candy, dark, ocean, retro, blueprint, forest, và journal. Tệp SVG được tạo không yêu cầu cài đặt Graphviz.

Metadata của Node#

Chỉnh sửa DISPLAY_CONFIG trong main.py của YAML2ModelGraph để chọn các chi tiết xuất hiện trong mỗi node:

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

Tên module YAML tùy chỉnh được hiển thị tự động. Màu sắc, hình dạng và kiểu chữ của chủ đề được cấu hình trong themes.py của công cụ.

Hạn chế#

YAML2ModelGraph căn chỉnh đặc biệt các head Detect tiêu chuẩn. Các head tác vụ khác như OBB, Pose, và Segment hiện được hiển thị dưới dạng các node làn neck chung. Xem kho lưu trữ YAML2ModelGraph để biết chi tiết về hỗ trợ và cách sử dụng hiện tại.