kf_config.json

{
  "library_name": "kerasformers",
  "kerasformers_version": "1.1.3",
  "model_module": "kerasformers.models.sam2",
  "model_class": "SAM2PromptableSegment",
  "variant": "sam2_hiera_base_plus",
  "weights": "model.weights.h5",
  "model_type": "sam2",
  "hidden_dim": 112,
  "blocks_per_stage": [
    2,
    3,
    16,
    3
  ],
  "embed_dim_per_stage": [
    112,
    224,
    448,
    896
  ],
  "num_attention_heads_per_stage": [
    2,
    4,
    8,
    16
  ],
  "window_size_per_stage": [
    8,
    4,
    14,
    7
  ],
  "global_attention_blocks": [
    12,
    16,
    20
  ],
  "backbone_channel_list": [
    896,
    448,
    224,
    112
  ],
  "window_pos_embed_bg_size": [
    14,
    14
  ],
  "num_multimask_outputs": 3,
  "include_box_input": false,
  "include_mask_input": false,
  "multimask_output": true,
  "image_size": 1024
}