kf_config.json

{
  "library_name": "kerasformers",
  "kerasformers_version": "1.2.1",
  "model_module": "kerasformers.models.mask2former",
  "model_class": "Mask2FormerUniversalSegment",
  "variant": "mask2former-swin-tiny-coco-instance",
  "weights": "model.weights.h5",
  "schema_version": 2,
  "weight_dtype": "float32",
  "model_type": "mask2former",
  "vision_config": {
    "backbone_embed_dim": 96,
    "backbone_depths": [
      2,
      2,
      6,
      2
    ],
    "backbone_num_heads": [
      3,
      6,
      12,
      24
    ],
    "backbone_window_size": 7,
    "hidden_dim": 256,
    "mask_feature_size": 256,
    "encoder_num_layers": 6,
    "encoder_ffn_dim": 1024,
    "decoder_num_layers": 9,
    "decoder_ffn_dim": 2048,
    "num_heads": 8,
    "num_queries": 100,
    "num_classes": 80,
    "image_size": 384
  }
}