File size: 946 Bytes
36e7d8d 7cd343a 36e7d8d | 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 | {
"model_type": "sam",
"architectures": [
"SamForSegmentation"
],
"initializer_range": 0.02,
"mask_decoder_config": {
"mask_input_channels": 16,
"hidden_size": 256,
"iou_head_depth": 3,
"iou_head_hidden_dim": 256,
"num_multimask_outputs": 3
},
"prompt_encoder_config": {
"embed_dim": 256,
"image_embedding_size": [
64,
64
],
"input_image_size": [
1024,
1024
],
"mask_in_channels": [
16,
32,
64,
128
],
"no_mask_embed": 256,
"num_point_embeddings": 4
},
"image_encoder": {
"type": "tiny_vit",
"embed_dim": 96,
"depth": [
2,
2,
4,
2
],
"num_heads": [
3,
6,
12,
24
],
"window_size": 7,
"mlp_ratio": 4.0,
"drop_path_rate": 0.1,
"img_size": 1024,
"patch_size": 16,
"out_chans": 256
},
"image_size": 1024,
"num_classes": 1
} |