Hugging Face's logo

HichTala
/
diffusiondet-dota

Object Detection
Transformers
Safetensors
diffusiondet
custom_code
Model card Files
xet
Community
diffusiondet-dota
File size: 2,490 Bytes 
e6a6ef1
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
 
{
  "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
}