timm documentation

DenseNet

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

DenseNet

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

شبكات التلافيف ذات الروابط الكثيفة، حيث تستقبل كل طبقة داخل كتلة كثيفة المخرجات المدمجة لجميع الطبقات السابقة لها في الكتلة. وقد تم تكييفها من torchvision، حيث يستخدم densenetblur عملية blur pool للعينة المنخفضة المضادة للتشويش.

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

الاستخدام

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

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

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

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

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

البنى

البنيةالمعاملات (مليون)GMACsحجم الإدخالالأوزان المُسبقة التدريب
densenet1217.982.872242
densenet16128.687.792241
densenet16914.153.402241
densenet20120.014.342241
densenet264d72.7413.572240
densenetblur121d8.003.112241

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

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

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

تُجمَّع الأوزان حسب المصدر: الورقة البحثية أو الأصل المذكور في الإعداد المسبق، وإلا منظمة على Hub غير timm، وإلا الجزء الأول من وسم التدريب المسبق (architecture.tag). تُدرج المصادر التي تحتوي على وزن واحد تحت other.

وسوم tv

الوزنحجم الإدخالأعلى 1 (%)حجم التقييمفئاترخصةالمركز
densenet121.tv_in1k22474.7502241000apache-2.0timm
densenet161.tv_in1k22477.3842241000apache-2.0timm
densenet169.tv_in1k22475.9222241000apache-2.0timm
densenet201.tv_in1k22477.2842241000apache-2.0timm

وسوم ra

الوزنحجم الإدخالأعلى 1 (%)حجم التقييمفئاترخصةالمركز
densenet121.ra_in1k224 (اختبار 288)76.4862881000apache-2.0timm
densenetblur121d.ra_in1k224 (اختبار 288)77.3202881000apache-2.0timm
Update on GitHub