timm documentation

BEiT

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

BEiT

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

تحويلات الرؤية BEiT وBEiT-v2، مُسبقة التدريب عبر نمذجة الصور المقنّعة. يتنبأ BEiT بالرموز البصرية المنفصلة للبقع المقنّعة، بينما يستخدم BEiT-v2 مُرمّزًا بصريًا مُكمّمًا متجهيًا للأهداف. الأوزان هنا هي أوزان تصنيف مُعدّلة بدقة.

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

الاستخدام

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

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

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

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

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

البنى

البنيةالمعاملات (مليون)GMACsحجم الإدخالالأوزان المُسبقة التدريب
beit_base_patch16_22486.5317.582242
beit_base_patch16_38486.7455.543841
beit_large_patch16_224304.4361.602242
beit_large_patch16_384305.00191.213841
beit_large_patch16_512305.67362.245121
beitv2_base_patch16_22486.5317.582243
beitv2_large_patch16_224304.4361.602243

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

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

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

الوزنحجم الإدخالأعلى 1 (%)حجم التقييمفئاترخصةالمركز
beit_base_patch16_224.in22k_ft_in22k224--21841apache-2.0timm
beit_base_patch16_224.in22k_ft_in22k_in1k22485.2382241000apache-2.0timm
beit_base_patch16_384.in22k_ft_in22k_in1k38486.8143841000apache-2.0timm
beit_large_patch16_224.in22k_ft_in22k224--21841apache-2.0timm
beit_large_patch16_224.in22k_ft_in22k_in1k22487.4862241000apache-2.0timm
beit_large_patch16_384.in22k_ft_in22k_in1k38488.3803841000apache-2.0timm
beit_large_patch16_512.in22k_ft_in22k_in1k51288.5765121000apache-2.0timm
beitv2_base_patch16_224.in1k_ft_in1k22485.5962241000apache-2.0timm
beitv2_base_patch16_224.in1k_ft_in22k224--21841apache-2.0timm
beitv2_base_patch16_224.in1k_ft_in22k_in1k22486.4942241000apache-2.0timm
beitv2_large_patch16_224.in1k_ft_in1k22487.4142241000apache-2.0timm
beitv2_large_patch16_224.in1k_ft_in22k224--21841apache-2.0timm
beitv2_large_patch16_224.in1k_ft_in22k_in1k22488.4062241000apache-2.0timm

الاستشهاد

@inproceedings{beit,
title={{BEiT}: {BERT} Pre-Training of Image Transformers},
author={Hangbo Bao and Li Dong and Songhao Piao and Furu Wei},
booktitle={International Conference on Learning Representations},
year={2022},
url={https://openreview.net/forum?id=p-BhZSz59o4}
}

@article{beitv2,
title={{BEiT v2}: Masked Image Modeling with Vector-Quantized Visual Tokenizers},
author={Zhiliang Peng and Li Dong and Hangbo Bao and Qixiang Ye and Furu Wei},
year={2022},
eprint={2208.06366},
archivePrefix={arXiv},
primaryClass={cs.CV}
}
Update on GitHub