YAML2ModelGraph ile YOLO Mimarilerini Görselleştir#

YAML2ModelGraph, Ultralytics YOLO model YAML dosyalarını SVG mimari diyagramlarına dönüştüren bir topluluk aracıdır. Özel modelleri belgelemene veya bunların backbone, neck ve head yapılarını incelemene yardımcı olabilir.

Kurulum#

Projeyi klonla ve PyYAML bağımlılığını yükle:

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

Kullanım#

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

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

Ayrı ölçeğe özgü head düğümleri çizmek için --head multi kullan veya dahil edilen temalardan birini seç:

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 bulunur. Oluşturulan SVG için Graphviz kurulumu gerekmez.

Düğüm Meta Verileri#

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

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 head düğümlerini özel olarak hizalar. OBB, Pose ve Segment gibi diğer görev head'leri şu anda genel neck şeridi düğümleri olarak işlenir. Güncel destek ve kullanım ayrıntıları için YAML2ModelGraph deposuna bak.

SSS#