timm documentation

RepGhost

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

RepGhost

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

يستبدل RepGhostNet دمج الميزات في وحدة Ghost بإعادة استخدام الميزات المعاد ترميزها، حيث تُدمج في مسار تجميع واحد عند الاستدلال.

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

الاستخدام

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

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

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

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

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

البنى

البنيةالمعاملات (مليون)GMACsحجم الإدخالالأوزان المُسبقة التدريب
repghostnet_0502.310.052241
repghostnet_0582.540.062241
repghostnet_0803.270.102241
repghostnet_1004.060.152241
repghostnet_1114.520.182241
repghostnet_1305.460.242241
repghostnet_1506.550.312241
repghostnet_2009.770.532241

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

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

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

الوزنحجم الإدخالأعلى 1 (%)حجم التقييمفئاترخصةالمركز
repghostnet_050.in1k22466.9782241000mittimm
repghostnet_058.in1k22468.9382241000mittimm
repghostnet_080.in1k22472.2402241000mittimm
repghostnet_100.in1k22474.2082241000mittimm
repghostnet_111.in1k22475.0602241000mittimm
repghostnet_130.in1k22476.3782241000mittimm
repghostnet_150.in1k22477.4742241000mittimm
repghostnet_200.in1k22478.8022241000mittimm
Update on GitHub