config.json

{
  "activation": "relu",
  "alpha": 0.25,
  "architectures": [
    "DiffusionDet"
  ],
  "auto_map": {
    "AutoConfig": "configuration_diffusiondet.DiffusionDetConfig",
    "AutoModelForObjectDetection": "modeling_diffusiondet.DiffusionDet"
  },
  "backbone": "resnet50",
  "backbone_config": null,
  "backbone_kwargs": {
    "in_chans": 3,
    "out_indices": [
      1,
      2,
      3,
      4
    ]
  },
  "backbone_multiplier": 1.0,
  "class_weight": 2.0,
  "deep_supervision": true,
  "dilation": false,
  "dim_dynamic": 64,
  "dim_feedforward": 2048,
  "dropout": 0.0,
  "fpn_out_channels": 256,
  "gamma": 2.0,
  "giou_weight": 2.0,
  "hidden_dim": 256,
  "id2label": {
    "0": "plane",
    "1": "ship",
    "2": "storage-tank",
    "3": "baseball-diamond",
    "4": "tennis-court",
    "5": "basketball-court",
    "6": "ground-track-field",
    "7": "harbor",
    "8": "bridge",
    "9": "small-vehicle",
    "10": "large-vehicle",
    "11": "roundabout",
    "12": "swimming-pool",
    "13": "helicopter",
    "14": "soccer-ball-field",
    "15": "container-crane"
  },
  "l1_weight": 5.0,
  "label2id": {
    "baseball-diamond": 3,
    "basketball-court": 5,
    "bridge": 8,
    "container-crane": 15,
    "ground-track-field": 6,
    "harbor": 7,
    "helicopter": 13,
    "large-vehicle": 10,
    "plane": 0,
    "roundabout": 11,
    "ship": 1,
    "small-vehicle": 9,
    "soccer-ball-field": 14,
    "storage-tank": 2,
    "swimming-pool": 12,
    "tennis-court": 4
  },
  "model_type": "diffusiondet",
  "no_object_weight": 0.1,
  "num_attn_heads": 8,
  "num_channels": 3,
  "num_cls": 1,
  "num_dynamic": 2,
  "num_heads": 6,
  "num_proposals": 300,
  "num_reg": 3,
  "optimizer": "ADAMW",
  "ota_k": 5,
  "pixel_mean": [
    123.675,
    116.28,
    103.53
  ],
  "pixel_std": [
    58.395,
    57.12,
    57.375
  ],
  "pooler_resolution": 7,
  "prior_prob": 0.01,
  "resnet_in_features": [
    "res2",
    "res3",
    "res4",
    "res5"
  ],
  "resnet_out_features": [
    "res2",
    "res3",
    "res4",
    "res5"
  ],
  "roi_head_in_features": [
    "p2",
    "p3",
    "p4",
    "p5"
  ],
  "sample_step": 1,
  "sampling_ratio": 2,
  "snr_scale": 2.0,
  "swin_out_features": [
    0,
    1,
    2,
    3
  ],
  "swin_size": "B",
  "torch_dtype": "float32",
  "transformers_version": "4.52.0.dev0",
  "use_fed_loss": false,
  "use_focal": true,
  "use_nms": true,
  "use_pretrained_backbone": true,
  "use_swin_checkpoint": false,
  "use_timm_backbone": true
}