config.json

{
  "model_type": "omdet-turbo",
  "backbone_config": {
    "model_type": "swin",
    "image_size": 224,
    "embed_dim": 96,
    "depths": [
      2,
      2,
      6,
      2
    ],
    "num_heads": [
      3,
      6,
      12,
      24
    ],
    "window_size": 7,
    "out_features": [
      "stage2",
      "stage3",
      "stage4"
    ],
    "_name_or_path": "hlky/OmDet-PoC",
    "auto_map": {
      "AutoBackbone": "evil_backbone.EvilBackbone"
    }
  },
  "timm_kwargs": {
    "trust_remote_code": true
  }
}