timm documentation

ConvMixer

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

ConvMixer

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

يضمّن ConvMixer الصورة على شكل رقعات ويحافظ على ذلك الدقة طوال العملية، متناوبًا بين التحويلات العميقة (الخلط المكاني) والتحويلات النقطية (الخلط بين القنوات).

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

الاستخدام

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

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

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

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

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

البنى

البنيةالمعاملات (مليون)GMACsحجم الإدخالالأوزان المُسبقة التدريب
convmixer_768_3221.1119.552241
convmixer_1024_20_ks9_p1424.385.552241
convmixer_1536_2051.6348.682241

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

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

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

الوزنحجم الإدخالأعلى 1 (%)حجم التقييمفئاترخصةالمركز
convmixer_768_32.in1k22480.1582241000mittimm
convmixer_1024_20_ks9_p14.in1k22476.9442241000mittimm
convmixer_1536_20.in1k22481.3822241000mittimm
Update on GitHub