kf_config.json

{
  "library_name": "kerasformers",
  "kerasformers_version": "1.1.3",
  "model_module": "kerasformers.models.sam2",
  "model_class": "SAM2PromptableSegment",
  "variant": "sam2_hiera_tiny",
  "weights": "model.weights.h5",
  "model_type": "sam2",
  "hidden_dim": 96,
  "blocks_per_stage": [
    1,
    2,
    7,
    2
  ],
  "embed_dim_per_stage": [
    96,
    192,
    384,
    768
  ],
  "num_attention_heads_per_stage": [
    1,
    2,
    4,
    8
  ],
  "window_size_per_stage": [
    8,
    4,
    14,
    7
  ],
  "global_attention_blocks": [
    5,
    7,
    9
  ],
  "backbone_channel_list": [
    768,
    384,
    192,
    96
  ],
  "window_pos_embed_bg_size": null,
  "num_multimask_outputs": 3,
  "include_box_input": false,
  "include_mask_input": false,
  "multimask_output": true,
  "image_size": 1024
}