timm documentation
DeiT
DeiT
Kategori: ViT · Kod: timm/models/deit.py · Mimariler: 16 · Önceden eğitilmiş ağırlıklar: 24 (24'ü ImageNet-1k sonuçlarıyla)
Veri verimli Görüntü Dönüşümcüleri (DeiT), yalnızca ImageNet-1k üzerinde eğitilen, isteğe bağlı olarak bir CNN öğretmeninden öğrenen bir damıtma jetonu içeren Görüntü Dönüşümcüleridir. DeiT III modelleri, geliştirilmiş bir tarif ile eğitilir.
Makaleler
Kullanım
Önceden eğitilmiş ağırlıkları yükleme, görüntü ön işleme ve ince ayar için Hızlı Başlangıç bölümüne, modelleri omurga olarak kullanmak için ise Özellik Çıkarımı bölümüne bakın.
>>> import timm
>>> model = timm.create_model('deit_small_distilled_patch16_224.fb_in1k', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)Model API desteği
| Özellik | VisionTransformer | VisionTransformerDistilled |
|---|---|---|
features_only=True özellik haritaları | ✓ (adım 16, 16, 16) | ✓ (adım 16, 16, 16) |
forward_intermediates() | ✓ | ✓ |
prune_intermediate_layers() | ✓ | ✓ |
| Herhangi bir giriş boyutu, değişiklik gerekmez | ✗ | ✗ |
set_input_size() | ✓ | ✓ |
dynamic_img_size=True | ✓ | ✓ |
set_grad_checkpointing() | ✓ | ✓ |
Katman bazlı LR azaltma (group_matcher()) | ✓ | ✓ |
reset_classifier() | ✓ | ✓ |
✓ destekleniyor, ✓ (kare) yalnızca kare boyutlarda çalışır, ✗ desteklenmiyor, – gerekli değil (model zaten herhangi bir giriş boyutunu kabul eder). Her özellik, rastgele başlatılmış bir model üzerinde çalıştırılarak kontrol edilir (VisionTransformer 224 için deit_tiny_patch16_224, VisionTransformerDistilled 224 için deit_tiny_distilled_patch16_224), aynı sınıftaki diğer mimarilerin aynı şekilde davranması beklenir. Ayrıntılar için Model API sayfasına bakın.
Mimariler
| Mimari | Sınıf | Parametreler (M) | GMAC'lar | Girdi boyutu | Önceden eğitilmiş ağırlıklar |
|---|---|---|---|---|---|
deit3_base_patch16_224 | VisionTransformer | 86.59 | 17.58 | 224 | 2 |
deit3_base_patch16_384 | VisionTransformer | 86.88 | 55.54 | 384 | 2 |
deit3_huge_patch14_224 | VisionTransformer | 632.13 | 167.40 | 224 | 2 |
deit3_large_patch16_224 | VisionTransformer | 304.37 | 61.60 | 224 | 2 |
deit3_large_patch16_384 | VisionTransformer | 304.76 | 191.21 | 384 | 2 |
deit3_medium_patch16_224 | VisionTransformer | 38.85 | 8.00 | 224 | 2 |
deit3_small_patch16_224 | VisionTransformer | 22.06 | 4.61 | 224 | 2 |
deit3_small_patch16_384 | VisionTransformer | 22.21 | 15.52 | 384 | 2 |
deit_base_distilled_patch16_224 | VisionTransformerDistilled | 87.34 | 17.68 | 224 | 1 |
deit_base_distilled_patch16_384 | VisionTransformerDistilled | 87.63 | 55.65 | 384 | 1 |
deit_base_patch16_224 | VisionTransformer | 86.57 | 17.58 | 224 | 1 |
deit_base_patch16_384 | VisionTransformer | 86.86 | 55.54 | 384 | 1 |
deit_small_distilled_patch16_224 | VisionTransformerDistilled | 22.44 | 4.63 | 224 | 1 |
deit_small_patch16_224 | VisionTransformer | 22.05 | 4.61 | 224 | 1 |
deit_tiny_distilled_patch16_224 | VisionTransformerDistilled | 5.91 | 1.27 | 224 | 1 |
deit_tiny_patch16_224 | VisionTransformer | 5.72 | 1.26 | 224 | 1 |
GMAC'lar ve parametre sayıları, her mimarinin varsayılan giriş boyutu ve sınıflandırıcısı içindir ve benchmark sonuçlarından alınmıştır.
Önceden eğitilmiş ağırlıklar
Top-1, belirtilen değerlendirme boyutunda results-imagenet.csv dosyasındaki ImageNet-1k doğrulama doğruluğudur; ağırlığın ImageNet-1k sonucu yoksa (ör. ön eğitim veya görsel-metin ağırlıkları) bu değer "-" olarak gösterilir. Classes, ön eğitilmiş sınıflandırıcının boyutudur; sınıflandırıcısı olmayan ağırlıklar için bu değer "none"dur.
| Ağırlık | Girdi boyutu | En iyi 1 (%) | Değerlendirme boyutu | Sınıflar | Lisans | Hub |
|---|---|---|---|---|---|---|
deit3_base_patch16_224.fb_in1k | 224 | 83.800 | 224 | 1000 | apache-2.0 | timm |
deit3_base_patch16_224.fb_in22k_ft_in1k | 224 | 85.728 | 224 | 1000 | apache-2.0 | timm |
deit3_base_patch16_384.fb_in1k | 384 | 85.088 | 384 | 1000 | apache-2.0 | timm |
deit3_base_patch16_384.fb_in22k_ft_in1k | 384 | 86.742 | 384 | 1000 | apache-2.0 | timm |
deit3_huge_patch14_224.fb_in1k | 224 | 85.204 | 224 | 1000 | apache-2.0 | timm |
deit3_huge_patch14_224.fb_in22k_ft_in1k | 224 | 87.180 | 224 | 1000 | apache-2.0 | timm |
deit3_large_patch16_224.fb_in1k | 224 | 84.772 | 224 | 1000 | apache-2.0 | timm |
deit3_large_patch16_224.fb_in22k_ft_in1k | 224 | 86.990 | 224 | 1000 | apache-2.0 | timm |
deit3_large_patch16_384.fb_in1k | 384 | 85.820 | 384 | 1000 | apache-2.0 | timm |
deit3_large_patch16_384.fb_in22k_ft_in1k | 384 | 87.736 | 384 | 1000 | apache-2.0 | timm |
deit3_medium_patch16_224.fb_in1k | 224 | 83.090 | 224 | 1000 | apache-2.0 | timm |
deit3_medium_patch16_224.fb_in22k_ft_in1k | 224 | 84.550 | 224 | 1000 | apache-2.0 | timm |
deit3_small_patch16_224.fb_in1k | 224 | 81.382 | 224 | 1000 | apache-2.0 | timm |
deit3_small_patch16_224.fb_in22k_ft_in1k | 224 | 83.082 | 224 | 1000 | apache-2.0 | timm |
deit3_small_patch16_384.fb_in1k | 384 | 83.432 | 384 | 1000 | apache-2.0 | timm |
deit3_small_patch16_384.fb_in22k_ft_in1k | 384 | 84.846 | 384 | 1000 | apache-2.0 | timm |
deit_base_distilled_patch16_224.fb_in1k | 224 | 83.384 | 224 | 1000 | apache-2.0 | timm |
deit_base_distilled_patch16_384.fb_in1k | 384 | 85.438 | 384 | 1000 | apache-2.0 | timm |
deit_base_patch16_224.fb_in1k | 224 | 81.980 | 224 | 1000 | apache-2.0 | timm |
deit_base_patch16_384.fb_in1k | 384 | 83.108 | 384 | 1000 | apache-2.0 | timm |
deit_small_distilled_patch16_224.fb_in1k | 224 | 81.204 | 224 | 1000 | apache-2.0 | timm |
deit_small_patch16_224.fb_in1k | 224 | 79.856 | 224 | 1000 | apache-2.0 | timm |
deit_tiny_distilled_patch16_224.fb_in1k | 224 | 74.538 | 224 | 1000 | apache-2.0 | timm |
deit_tiny_patch16_224.fb_in1k | 224 | 72.190 | 224 | 1000 | apache-2.0 | timm |