timm documentation

RepViT

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

RepViT

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

RepViT هو شبكة عصبية تلافيفية (CNN) مصممة للأجهزة المحمولة، يجلب خيارات التصميم الخاصة بتحويلات الرؤية الخفيفة إلى كتل بأسلوب MobileNet-V3، مع إعادة التمثيل الهيكلي.

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

الاستخدام

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

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

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

ميزةRepVit
خرائط الميزات 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()✓

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

البنى

البنيةالمعاملات (مليون)GMACsحجم الإدخالالأوزان المُسبقة التدريب
repvit_m0_95.070.822242
repvit_m15.070.822241
repvit_m1_06.811.112242
repvit_m1_18.241.342242
repvit_m1_514.052.272242
repvit_m28.241.342241
repvit_m2_322.934.522242
repvit_m310.121.862241

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

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

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

الوزنحجم الإدخالأعلى 1 (%)حجم التقييمفئاترخصةالمركز
repvit_m0_9.dist_300e_in1k22478.6522241000apache-2.0timm
repvit_m0_9.dist_450e_in1k22479.0582241000apache-2.0timm
repvit_m1.dist_in1k22478.5422241000apache-2.0timm
repvit_m1_0.dist_300e_in1k22480.1522241000apache-2.0timm
repvit_m1_0.dist_450e_in1k22480.4302241000apache-2.0timm
repvit_m1_1.dist_300e_in1k22480.8262241000apache-2.0timm
repvit_m1_1.dist_450e_in1k22481.3142241000apache-2.0timm
repvit_m1_5.dist_300e_in1k22482.3822241000apache-2.0timm
repvit_m1_5.dist_450e_in1k22482.5202241000apache-2.0timm
repvit_m2.dist_in1k22480.4682241000apache-2.0timm
repvit_m2_3.dist_300e_in1k22483.5242241000apache-2.0timm
repvit_m2_3.dist_450e_in1k22483.7462241000apache-2.0timm
repvit_m3.dist_in1k22481.4942241000apache-2.0timm

الاستشهاد

@misc{wang2023repvit,
      title={RepViT: Revisiting Mobile CNN From ViT Perspective},
      author={Ao Wang and Hui Chen and Zijia Lin and Hengjun Pu and Guiguang Ding},
      year={2023},
      eprint={2307.09283},
      archivePrefix={arXiv},
      primaryClass={cs.CV}
}
Update on GitHub