Hugging Face's logo

microsoft
/
STELLAR

Image Feature Extraction
stellar
vision
self-supervised-learning
representation-learning
sparse-tokens
vision-transformer
Model card Files
xet
 Community
STELLAR
File size: 2,327 Bytes
6794be9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
{
  "model_type": "stellar",
  "library_name": "stellar",
  "code": "https://github.com/microsoft/STELLAR",
  "default": "stellar-b16",
  "models": {
    "stellar-b16": {
      "weights": "stellar-b16.safetensors",
      "category": "main",
      "backbone": "facebook/vit-mae-base",
      "image_size": 224,
      "patch_size": 16,
      "embed_dim": 768,
      "num_sparse_tokens": 16,
      "num_decoder_layers": 6,
      "spatial_temp": 0.06,
      "do_recon": true,
      "do_clustering": true,
      "do_cls": true,
      "num_clusters": 16384,
      "recon_type": "vq"
    },
    "stellar-l16": {
      "weights": "stellar-l16.safetensors",
      "category": "main",
      "backbone": "facebook/vit-mae-large",
      "image_size": 224,
      "patch_size": 16,
      "embed_dim": 1024,
      "num_sparse_tokens": 16,
      "num_decoder_layers": 6,
      "spatial_temp": 0.06,
      "do_recon": true,
      "do_clustering": true,
      "do_cls": true,
      "num_clusters": 16384,
      "recon_type": "vq"
    },
    "stellar-h16": {
      "weights": "stellar-h16.safetensors",
      "category": "main",
      "backbone": "facebook/vit-mae-huge",
      "image_size": 224,
      "patch_size": 14,
      "embed_dim": 1280,
      "num_sparse_tokens": 16,
      "num_decoder_layers": 6,
      "spatial_temp": 0.06,
      "do_recon": true,
      "do_clustering": true,
      "do_cls": true,
      "num_clusters": 16384,
      "recon_type": "vq"
    },
    "stellar-b8": {
      "weights": "stellar-b8.safetensors",
      "category": "ablation",
      "backbone": "facebook/vit-mae-base",
      "image_size": 224,
      "patch_size": 16,
      "embed_dim": 768,
      "num_sparse_tokens": 8,
      "num_decoder_layers": 6,
      "spatial_temp": 0.06,
      "do_recon": true,
      "do_clustering": true,
      "do_cls": true,
      "num_clusters": 16384,
      "recon_type": "vq"
    },
    "stellar-b24": {
      "weights": "stellar-b24.safetensors",
      "category": "ablation",
      "backbone": "facebook/vit-mae-base",
      "image_size": 224,
      "patch_size": 16,
      "embed_dim": 768,
      "num_sparse_tokens": 24,
      "num_decoder_layers": 6,
      "spatial_temp": 0.06,
      "do_recon": true,
      "do_clustering": true,
      "do_cls": true,
      "num_clusters": 16384,
      "recon_type": "vq"
    }
  }
}