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 phần thân (backbone), cổ (neck) và đầu (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 phần đầu 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.

Siêu dữ liệu 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 mô-đun 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 phần đầu Detect tiêu chuẩn. Các phần đầu 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 cổ tổng quát. Xem kho lưu trữ YAML2ModelGraph để biết chi tiết hỗ trợ và cách sử dụng hiện tại.