رؤية YOLO 2026:

تصوير بنيات YOLO باستخدام YAML2ModelGraph#

YAML2ModelGraph هو أداة مجتمعية تحول ملفات YAML الخاصة بنماذج Ultralytics YOLO إلى رسوم تخطيطية لبنية SVG. يمكن أن تساعد في توثيق النماذج المخصصة أو فحص هيكل العمود الفقري (backbone)، والعنق (neck)، والرأس (head).

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

التثبيت#

استنسخ المشروع وقم بتثبيت تبعية PyYAML الخاصة به:

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

الاستخدام#

أنشئ رسماً تخطيطياً من مثال YOLO26 المرفق:

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

استخدم --head multi لرسم عقد رأس منفصلة خاصة بالحجم، أو حدد أحد المظاهر المضمنة:

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

تشمل المظاهر المتاحة paper، وpaper_ryb، وcandy، وdark، وocean، وretro، وblueprint، وforest، وjournal. ملف SVG الناتج لا يتطلب تثبيت Graphviz.

بيانات وصفية للعقدة#

قم بتحرير DISPLAY_CONFIG في main.py الخاص بـ YAML2ModelGraph لاختيار التفاصيل التي تظهر في كل عقدة:

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

يتم عرض أسماء وحدة YAML المخصصة تلقائياً. يتم تكوين ألوان المظهر، والأشكال، والطباعة في themes.py الخاص بالأداة.

القيود#

تقوم أداة YAML2ModelGraph بمحاذاة رؤوس Detect القياسية بشكل خاص. رؤوس المهام الأخرى مثل OBB، وPose، وSegment يتم عرضها حالياً كعقد مسار عنق عامة. راجع مستودع YAML2ModelGraph للحصول على تفاصيل الدعم والاستخدام الحالية.

المساهمون
RARahulbiradar91

التعليقات