Hugging Face's logo

zeromodels
/
sam2_hiera_base_plus

Mask Generation
Keras
sam2
PyTorch
JAX
TensorFlow
zeromodels
image-segmentation
Model card Files
xet
 Community
IMvision12 commited on
Commit
349f2b5
·
verified ·
1 Parent(s): 30ead1c

Upgrade kf_config.json to v2 (nested config + dtype)

Browse files
Files changed (1) hide show
  1. kf_config.json +50 -46
kf_config.json CHANGED
@@ -1,54 +1,58 @@
1
  {
2
  "library_name": "kerasformers",
3
- "kerasformers_version": "1.1.3",
4
  "model_module": "kerasformers.models.sam2",
5
  "model_class": "SAM2PromptableSegment",
6
  "variant": "sam2_hiera_base_plus",
7
  "weights": "model.weights.h5",
 
 
8
  "model_type": "sam2",
9
- "hidden_dim": 112,
10
- "blocks_per_stage": [
11
- 2,
12
- 3,
13
- 16,
14
- 3
15
- ],
16
- "embed_dim_per_stage": [
17
- 112,
18
- 224,
19
- 448,
20
- 896
21
- ],
22
- "num_attention_heads_per_stage": [
23
- 2,
24
- 4,
25
- 8,
26
- 16
27
- ],
28
- "window_size_per_stage": [
29
- 8,
30
- 4,
31
- 14,
32
- 7
33
- ],
34
- "global_attention_blocks": [
35
- 12,
36
- 16,
37
- 20
38
- ],
39
- "backbone_channel_list": [
40
- 896,
41
- 448,
42
- 224,
43
- 112
44
- ],
45
- "window_pos_embed_bg_size": [
46
- 14,
47
- 14
48
- ],
49
- "num_multimask_outputs": 3,
50
- "include_box_input": false,
51
- "include_mask_input": false,
52
- "multimask_output": true,
53
- "image_size": 1024
 
 
54
  }
 
1
  {
2
  "library_name": "kerasformers",
3
+ "kerasformers_version": "1.2.1",
4
  "model_module": "kerasformers.models.sam2",
5
  "model_class": "SAM2PromptableSegment",
6
  "variant": "sam2_hiera_base_plus",
7
  "weights": "model.weights.h5",
8
+ "schema_version": 2,
9
+ "dtype": "float32",
10
  "model_type": "sam2",
11
+ "vision_config": {
12
+ "hidden_dim": 112,
13
+ "blocks_per_stage": [
14
+ 2,
15
+ 3,
16
+ 16,
17
+ 3
18
+ ],
19
+ "embed_dim_per_stage": [
20
+ 112,
21
+ 224,
22
+ 448,
23
+ 896
24
+ ],
25
+ "num_attention_heads_per_stage": [
26
+ 2,
27
+ 4,
28
+ 8,
29
+ 16
30
+ ],
31
+ "window_size_per_stage": [
32
+ 8,
33
+ 4,
34
+ 14,
35
+ 7
36
+ ],
37
+ "global_attention_blocks": [
38
+ 12,
39
+ 16,
40
+ 20
41
+ ],
42
+ "backbone_channel_list": [
43
+ 896,
44
+ 448,
45
+ 224,
46
+ 112
47
+ ],
48
+ "window_pos_embed_bg_size": [
49
+ 14,
50
+ 14
51
+ ],
52
+ "num_multimask_outputs": 3,
53
+ "include_box_input": false,
54
+ "include_mask_input": false,
55
+ "multimask_output": true,
56
+ "image_size": 1024
57
+ }
58
  }