timm documentation

TinyViT

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v1.0.30).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

TinyViT

الفئة: هجينة · الكود: timm/models/tiny_vit.py · الهندسات: 5 · الأوزان المُسبقة التدريب: 11 (8 منها بنتائج على ImageNet-1k)

TinyViT هو عائلة من محولات الرؤية الهرمية الصغيرة (مرحلة أولى تجميعية، ثم انتباه نافذة محلية)، تم تدريبها مسبقًا عن طريق التقطير السريع من نماذج معلمة كبيرة.

الأوراق البحثية

الاستخدام

راجع البدء السريع لتحميل الأوزان المُسبقة التدريب ومعالجة الصور مسبقًا وضبط الدقة، وراجع استخراج الميزات لاستخدام النماذج كعمود فقري.

>>> import timm
>>> model = timm.create_model('tiny_vit_21m_512.dist_in22k_ft_in1k', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)

دعم واجهة برمجة التطبيقات للنماذج

ميزةTinyVit
خرائط الميزات features_only=True✓ (خطوات 4، 8، 16، 32)
forward_intermediates()✓
prune_intermediate_layers()✓
أي حجم إدخال، دون الحاجة إلى أي تغييرات✓
set_input_size()–
dynamic_img_size=True–
set_grad_checkpointing()✓
تلاشي معدل التعلم على مستوى الطبقات (group_matcher())✓
reset_classifier()✓

✓ مدعوم، ✓ (مربع) يعمل مع الأحجام المربعة فقط، ✗ غير مدعوم، – غير مطلوب (يقبل النموذج بالفعل أي حجم إدخال). تم التحقق من ذلك بتشغيل كل ميزة على نموذج مُهيأ عشوائياً (TinyVit مع tiny_vit_5m_224 عند 224)، ومن المتوقع أن تتصرف البنى المعمارية الأخرى من نفس الفئة بنفس الطريقة. راجع صفحة واجهة برمجة النموذج للتفاصيل.

البنى

البنيةالمعاملات (مليون)GMACsحجم الإدخالالأوزان المُسبقة التدريب
tiny_vit_5m_22412.081.272243
tiny_vit_11m_22420.352.032243
tiny_vit_21m_22433.214.272243
tiny_vit_21m_38421.2213.723841
tiny_vit_21m_51221.2626.935121

تُحسب قيم GMACs وعدد المعلمات لحجم المدخلات الافتراضي والتصنيف الخاص بكل معمارية، وذلك بناءً على نتائج الاختبارات المرجعية.

الأوزان المُسبقة التدريب

Top-1 هو دقة التحقق على ImageNet-1k من results-imagenet.csv عند حجم التقييم المذكور، - إذا لم يكن للوزن نتيجة على ImageNet-1k (مثل أوزان ما قبل التدريب أو أوزان الصورة-النص). Classes هو حجم المصنِّف المُدرَّب مسبقًا، ولا يوجد شيء للأوزان التي لا تحتوي على مصنِّف.

الوزنحجم الإدخالأعلى 1 (%)حجم التقييمفئاترخصةالمركز
tiny_vit_5m_224.dist_in22k224--21841apache-2.0timm
tiny_vit_5m_224.dist_in22k_ft_in1k22480.8742241000apache-2.0timm
tiny_vit_5m_224.in1k22479.1962241000apache-2.0timm
tiny_vit_11m_224.dist_in22k224--21841apache-2.0timm
tiny_vit_11m_224.dist_in22k_ft_in1k22483.2322241000apache-2.0timm
tiny_vit_11m_224.in1k22481.5382241000apache-2.0timm
tiny_vit_21m_224.dist_in22k224--21841apache-2.0timm
tiny_vit_21m_224.dist_in22k_ft_in1k22485.0882241000apache-2.0timm
tiny_vit_21m_224.in1k22483.2682241000apache-2.0timm
tiny_vit_21m_384.dist_in22k_ft_in1k38486.0883841000apache-2.0timm
tiny_vit_21m_512.dist_in22k_ft_in1k51286.4445121000apache-2.0timm
Update on GitHub