YOLO Vision 2026:

YOLO Mimarilerini YAML2ModelGraph ile 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 bunların omurga, boyun ve başlık yapısını incelemenize yardımcı olabilir.

Single-head YOLO architecture rendered by YAML2ModelGraph Multi-head YOLO architecture rendered by YAML2ModelGraph

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 detayları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ı özel olarak hizalar. OBB, Pose ve Segment gibi diğer görev başlıkları şu anda genel boyun şeridi düğümleri olarak işlenir. Mevcut destek ve kullanım detayları için YAML2ModelGraph repository sayfasına bakın.

Katkıda Bulunanlar
RARahulbiradar91

Yorumlar