timm documentation
HieraDet (SAM 2)
HieraDet (SAM 2)
Kategori: ViT · Kod: timm/models/hieradet_sam2.py · Mimariler: 5 · Önceden eğitilmiş ağırlıklar: 8 (ImageNet-1k sonuçları olan 0)
Segment Anything 2 (SAM 2)'nin pencere tabanlı dikkat ve pencere tabanlı konum gömme özelliklerine sahip, Hiera tabanlı görüntü kodlayıcısı olan HieraDet.
Makaleler
- SAM 2: Görsellerde ve Videolarda Her Şeyi Segmente Et
- Hiera: Süsleri Olmayan Hiyerarşik Görüntü Dönüştürücüsü
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('sam2_hiera_base_plus.fb_r896', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)Model API desteği
✓ 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 (HieraDet, 256 boyutunda sam2_hiera_tiny ile); aynı sınıftaki diğer mimarilerin de aynı şekilde davranması beklenir. Ayrıntılar için Model API sayfasına bakın.
Mimariler
| Mimari | Parametreler (M) | GMAC'lar | Girdi boyutu | Önceden eğitilmiş ağırlıklar |
|---|---|---|---|---|
hieradet_small | 34.73 | 8.51 | 256 | 0 |
sam2_hiera_base_plus | 68.68 | 227.48 | 896 | 2 |
sam2_hiera_large | 212.15 | 907.48 | 1024 | 2 |
sam2_hiera_small | 33.95 | 123.99 | 896 | 2 |
sam2_hiera_tiny | 26.85 | 99.86 | 896 | 2 |
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 |
|---|---|---|---|---|---|---|
sam2_hiera_base_plus.fb_r896 | 896 | - | - | none | apache-2.0 | timm |
sam2_hiera_base_plus.fb_r896_2pt1 | 896 | - | - | none | apache-2.0 | timm |
sam2_hiera_large.fb_r1024 | 1024 | - | - | none | apache-2.0 | timm |
sam2_hiera_large.fb_r1024_2pt1 | 1024 | - | - | none | apache-2.0 | timm |
sam2_hiera_small.fb_r896 | 896 | - | - | none | apache-2.0 | timm |
sam2_hiera_small.fb_r896_2pt1 | 896 | - | - | none | apache-2.0 | timm |
sam2_hiera_tiny.fb_r896 | 896 | - | - | none | apache-2.0 | timm |
sam2_hiera_tiny.fb_r896_2pt1 | 896 | - | - | none | apache-2.0 | timm |