YOLO Vision 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

التعليقات