HichTala commited on
Commit
378d045
·
verified ·
1 Parent(s): 518e7ee

Upload DiffusionDet

Browse files
Files changed (2) hide show
  1. config.json +178 -3
  2. model.safetensors +3 -0
config.json CHANGED
@@ -1,11 +1,20 @@
1
  {
2
- "_class_name": "DiffusionDetConfig",
3
- "_diffusers_version": "0.32.0.dev0",
4
  "activation": "relu",
5
  "alpha": 0.25,
 
 
 
6
  "backbone": "resnet50",
7
  "backbone_config": null,
8
- "backbone_kwargs": null,
 
 
 
 
 
 
 
 
9
  "backbone_multiplier": 1.0,
10
  "class_weight": 2.0,
11
  "deep_supervision": true,
@@ -17,7 +26,171 @@
17
  "gamma": 2.0,
18
  "giou_weight": 2.0,
19
  "hidden_dim": 256,
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
20
  "l1_weight": 5.0,
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
21
  "no_object_weight": 0.1,
22
  "num_attn_heads": 8,
23
  "num_channels": 3,
@@ -68,6 +241,8 @@
68
  3
69
  ],
70
  "swin_size": "B",
 
 
71
  "use_fed_loss": false,
72
  "use_focal": true,
73
  "use_nms": true,
 
1
  {
 
 
2
  "activation": "relu",
3
  "alpha": 0.25,
4
+ "architectures": [
5
+ "DiffusionDet"
6
+ ],
7
  "backbone": "resnet50",
8
  "backbone_config": null,
9
+ "backbone_kwargs": {
10
+ "in_chans": 3,
11
+ "out_indices": [
12
+ 1,
13
+ 2,
14
+ 3,
15
+ 4
16
+ ]
17
+ },
18
  "backbone_multiplier": 1.0,
19
  "class_weight": 2.0,
20
  "deep_supervision": true,
 
26
  "gamma": 2.0,
27
  "giou_weight": 2.0,
28
  "hidden_dim": 256,
29
+ "id2label": {
30
+ "0": "LABEL_0",
31
+ "1": "LABEL_1",
32
+ "2": "LABEL_2",
33
+ "3": "LABEL_3",
34
+ "4": "LABEL_4",
35
+ "5": "LABEL_5",
36
+ "6": "LABEL_6",
37
+ "7": "LABEL_7",
38
+ "8": "LABEL_8",
39
+ "9": "LABEL_9",
40
+ "10": "LABEL_10",
41
+ "11": "LABEL_11",
42
+ "12": "LABEL_12",
43
+ "13": "LABEL_13",
44
+ "14": "LABEL_14",
45
+ "15": "LABEL_15",
46
+ "16": "LABEL_16",
47
+ "17": "LABEL_17",
48
+ "18": "LABEL_18",
49
+ "19": "LABEL_19",
50
+ "20": "LABEL_20",
51
+ "21": "LABEL_21",
52
+ "22": "LABEL_22",
53
+ "23": "LABEL_23",
54
+ "24": "LABEL_24",
55
+ "25": "LABEL_25",
56
+ "26": "LABEL_26",
57
+ "27": "LABEL_27",
58
+ "28": "LABEL_28",
59
+ "29": "LABEL_29",
60
+ "30": "LABEL_30",
61
+ "31": "LABEL_31",
62
+ "32": "LABEL_32",
63
+ "33": "LABEL_33",
64
+ "34": "LABEL_34",
65
+ "35": "LABEL_35",
66
+ "36": "LABEL_36",
67
+ "37": "LABEL_37",
68
+ "38": "LABEL_38",
69
+ "39": "LABEL_39",
70
+ "40": "LABEL_40",
71
+ "41": "LABEL_41",
72
+ "42": "LABEL_42",
73
+ "43": "LABEL_43",
74
+ "44": "LABEL_44",
75
+ "45": "LABEL_45",
76
+ "46": "LABEL_46",
77
+ "47": "LABEL_47",
78
+ "48": "LABEL_48",
79
+ "49": "LABEL_49",
80
+ "50": "LABEL_50",
81
+ "51": "LABEL_51",
82
+ "52": "LABEL_52",
83
+ "53": "LABEL_53",
84
+ "54": "LABEL_54",
85
+ "55": "LABEL_55",
86
+ "56": "LABEL_56",
87
+ "57": "LABEL_57",
88
+ "58": "LABEL_58",
89
+ "59": "LABEL_59",
90
+ "60": "LABEL_60",
91
+ "61": "LABEL_61",
92
+ "62": "LABEL_62",
93
+ "63": "LABEL_63",
94
+ "64": "LABEL_64",
95
+ "65": "LABEL_65",
96
+ "66": "LABEL_66",
97
+ "67": "LABEL_67",
98
+ "68": "LABEL_68",
99
+ "69": "LABEL_69",
100
+ "70": "LABEL_70",
101
+ "71": "LABEL_71",
102
+ "72": "LABEL_72",
103
+ "73": "LABEL_73",
104
+ "74": "LABEL_74",
105
+ "75": "LABEL_75",
106
+ "76": "LABEL_76",
107
+ "77": "LABEL_77",
108
+ "78": "LABEL_78",
109
+ "79": "LABEL_79"
110
+ },
111
  "l1_weight": 5.0,
112
+ "label2id": {
113
+ "LABEL_0": 0,
114
+ "LABEL_1": 1,
115
+ "LABEL_10": 10,
116
+ "LABEL_11": 11,
117
+ "LABEL_12": 12,
118
+ "LABEL_13": 13,
119
+ "LABEL_14": 14,
120
+ "LABEL_15": 15,
121
+ "LABEL_16": 16,
122
+ "LABEL_17": 17,
123
+ "LABEL_18": 18,
124
+ "LABEL_19": 19,
125
+ "LABEL_2": 2,
126
+ "LABEL_20": 20,
127
+ "LABEL_21": 21,
128
+ "LABEL_22": 22,
129
+ "LABEL_23": 23,
130
+ "LABEL_24": 24,
131
+ "LABEL_25": 25,
132
+ "LABEL_26": 26,
133
+ "LABEL_27": 27,
134
+ "LABEL_28": 28,
135
+ "LABEL_29": 29,
136
+ "LABEL_3": 3,
137
+ "LABEL_30": 30,
138
+ "LABEL_31": 31,
139
+ "LABEL_32": 32,
140
+ "LABEL_33": 33,
141
+ "LABEL_34": 34,
142
+ "LABEL_35": 35,
143
+ "LABEL_36": 36,
144
+ "LABEL_37": 37,
145
+ "LABEL_38": 38,
146
+ "LABEL_39": 39,
147
+ "LABEL_4": 4,
148
+ "LABEL_40": 40,
149
+ "LABEL_41": 41,
150
+ "LABEL_42": 42,
151
+ "LABEL_43": 43,
152
+ "LABEL_44": 44,
153
+ "LABEL_45": 45,
154
+ "LABEL_46": 46,
155
+ "LABEL_47": 47,
156
+ "LABEL_48": 48,
157
+ "LABEL_49": 49,
158
+ "LABEL_5": 5,
159
+ "LABEL_50": 50,
160
+ "LABEL_51": 51,
161
+ "LABEL_52": 52,
162
+ "LABEL_53": 53,
163
+ "LABEL_54": 54,
164
+ "LABEL_55": 55,
165
+ "LABEL_56": 56,
166
+ "LABEL_57": 57,
167
+ "LABEL_58": 58,
168
+ "LABEL_59": 59,
169
+ "LABEL_6": 6,
170
+ "LABEL_60": 60,
171
+ "LABEL_61": 61,
172
+ "LABEL_62": 62,
173
+ "LABEL_63": 63,
174
+ "LABEL_64": 64,
175
+ "LABEL_65": 65,
176
+ "LABEL_66": 66,
177
+ "LABEL_67": 67,
178
+ "LABEL_68": 68,
179
+ "LABEL_69": 69,
180
+ "LABEL_7": 7,
181
+ "LABEL_70": 70,
182
+ "LABEL_71": 71,
183
+ "LABEL_72": 72,
184
+ "LABEL_73": 73,
185
+ "LABEL_74": 74,
186
+ "LABEL_75": 75,
187
+ "LABEL_76": 76,
188
+ "LABEL_77": 77,
189
+ "LABEL_78": 78,
190
+ "LABEL_79": 79,
191
+ "LABEL_8": 8,
192
+ "LABEL_9": 9
193
+ },
194
  "no_object_weight": 0.1,
195
  "num_attn_heads": 8,
196
  "num_channels": 3,
 
241
  3
242
  ],
243
  "swin_size": "B",
244
+ "torch_dtype": "float32",
245
+ "transformers_version": "4.47.0.dev0",
246
  "use_fed_loss": false,
247
  "use_focal": true,
248
  "use_nms": true,
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d10cceefec40f9708688e70ee509c4e8b485de3d07a88e1933f24eb46b8b6634
3
+ size 443203280