YAML2ModelGraph ile YOLO Mimarilerini Görselleştirin#

YAML2ModelGraph, Ultralytics YOLO model YAML dosyalarını SVG mimari diyagramlarına dönüştüren bir topluluk aracıdır. Özel modelleri belgelemenize veya omurga, boyun ve başlık yapılarını incelemenize yardımcı olabilir.

Kurulum#

Projeyi klonlayın ve PyYAML bağımlılığını yükleyin:

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

Kullanım#

Dahil edilen YOLO26 örneğinden bir diyagram oluşturun:

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

Ölçeğe özel ayrı başlık düğümleri çizmek için --head multi kullanın veya dahil edilen temalardan birini seçin:

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

Kullanılabilir temalar arasında paper, paper_ryb, candy, dark, ocean, retro, blueprint, forest ve journal yer alır. Oluşturulan SVG, herhangi bir Graphviz kurulumu gerektirmez.

Düğüm Meta Verileri#

Her düğümde hangi ayrıntıların görüneceğini seçmek için YAML2ModelGraph'ın main.py dosyasındaki DISPLAY_CONFIG öğesini düzenleyin:

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

Özel YAML modül adları otomatik olarak işlenir. Tema renkleri, şekilleri ve tipografisi aracın themes.py dosyasında yapılandırılır.

Sınırlamalar#

YAML2ModelGraph, standart Detect başlıklarını özellikle hizalar. OBB, Pose ve Segment gibi diğer görev başlıkları şu anda genel boyun şeridi düğümleri olarak işlenmektedir. Güncel destek ve kullanım ayrıntıları için YAML2ModelGraph deposuna bakın.