YAML2ModelGraph ile YOLO Mimarilerini Görselleştirme#

YAML2ModelGraph, Ultralytics YOLO model YAML dosyalarını SVG mimari diyagramlarına dönüştüren bir topluluk aracıdır. Özel modelleri belgelendirmeye veya modellerin backbone, neck ve head yapılarını incelemeye 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'ın 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 görüntülenir. Tema renkleri, şekilleri ve tipografisi aracın themes.py dosyasında yapılandırılır.

Sınırlamalar#

YAML2ModelGraph, standart Detect head'lerini özel olarak hizalar. Segment, Pose ve OBB gibi diğer görev head'leri şu anda genel neck şeridi düğümleri olarak görüntülenir. Güncel destek ve kullanım ayrıntıları için YAML2ModelGraph deposuna bak.

Sık Sorulan Sorular#