Hugging Face's logo

Spaces:
LXT
/
OMG_Seg
Sleeping

App Files Community
4
OMG_Seg
File size: 1,184 Bytes 
b34d1d6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
 
meta_dict = {
    'vit_h': dict(
        encoder_embed_dim=1280,
        encoder_depth=32,
        encoder_num_heads=16,
        encoder_global_attn_indexes=[7, 15, 23, 31],
        # common
        prompt_embed_dim=256,
        image_size=1024,
        vit_patch_size=16,
        image_embedding_size=64
    ),
    'vit_l': dict(
        encoder_embed_dim=1024,
        encoder_depth=24,
        encoder_num_heads=16,
        encoder_global_attn_indexes=[5, 11, 17, 23],
        # common
        prompt_embed_dim=256,
        image_size=1024,
        vit_patch_size=16,
        image_embedding_size=64
    ),
    'vit_b': dict(
        encoder_embed_dim=768,
        encoder_depth=12,
        encoder_num_heads=12,
        encoder_global_attn_indexes=[2, 5, 8, 11],
        # common
        prompt_embed_dim=256,
        image_size=1024,
        vit_patch_size=16,
        image_embedding_size=64
    )
}

checkpoint_dict = {
    'vit_h': 'https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth',
    'vit_l': 'https://dl.fbaipublicfiles.com/segment_anything/sam_vit_l_0b3195.pth',
    'vit_b': 'https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth',
}