kf_config.json

{
  "library_name": "kerasformers",
  "kerasformers_version": "1.1.3",
  "model_module": "kerasformers.models.mask2former",
  "model_class": "Mask2FormerUniversalSegment",
  "variant": "mask2former-swin-tiny-coco-instance",
  "weights": "model.weights.h5",
  "model_type": "mask2former",
  "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
}