zm_config.json

{
  "library_name": "zeromodels",
  "zeromodels_version": "1.2.1",
  "model_module": "zeromodels.models.rf_detr",
  "model_class": "RFDETRDetect",
  "variant": "rfdetr-nano",
  "weights": "model.weights.h5",
  "schema_version": 2,
  "weight_dtype": "float32",
  "model_type": "rf_detr",
  "vision_config": {
    "hidden_dim": 256,
    "backbone_hidden_size": 384,
    "backbone_num_heads": 6,
    "backbone_num_layers": 12,
    "backbone_mlp_ratio": 4,
    "backbone_use_swiglu": false,
    "num_register_tokens": 0,
    "out_feature_indexes": [
      3,
      6,
      9,
      12
    ],
    "patch_size": 16,
    "num_windows": 2,
    "positional_encoding_size": 24,
    "resolution": 384,
    "dec_layers": 2,
    "sa_nheads": 8,
    "ca_nheads": 16,
    "dec_n_points": 2,
    "num_queries": 300,
    "num_classes": 91,
    "two_stage": true,
    "bbox_reparam": true,
    "lite_refpoint_refine": true,
    "group_detr": 13,
    "dim_feedforward": 2048,
    "image_size": null
  }
}