FoamoftheSea commited on
Commit
590e645
·
1 Parent(s): ab01808

upload config and model

Browse files
Files changed (2) hide show
  1. config.json +248 -0
  2. pytorch_model.bin +3 -0
config.json ADDED
@@ -0,0 +1,248 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "activation_dropout": 0.0,
3
+ "activation_function": "relu",
4
+ "architectures": [
5
+ "Multiformer"
6
+ ],
7
+ "attention_dropout": 0.0,
8
+ "auxiliary_loss": false,
9
+ "backbone": "pvt_v2",
10
+ "backbone_config": {
11
+ "_name_or_path": "",
12
+ "_out_features": [
13
+ "stem",
14
+ "stage1",
15
+ "stage2",
16
+ "stage3"
17
+ ],
18
+ "_out_indices": [
19
+ 0,
20
+ 1,
21
+ 2,
22
+ 3
23
+ ],
24
+ "add_cross_attention": false,
25
+ "architectures": null,
26
+ "attention_probs_dropout_prob": 0.0,
27
+ "attn_reduce": "SR",
28
+ "bad_words_ids": null,
29
+ "begin_suppress_tokens": null,
30
+ "bos_token_id": null,
31
+ "chunk_size_feed_forward": 0,
32
+ "cross_attention_hidden_size": null,
33
+ "decoder_start_token_id": null,
34
+ "depths": [
35
+ 2,
36
+ 2,
37
+ 2,
38
+ 2
39
+ ],
40
+ "diversity_penalty": 0.0,
41
+ "do_reduce_labels": true,
42
+ "do_sample": false,
43
+ "drop_path_rate": 0.0,
44
+ "early_stopping": false,
45
+ "encoder_no_repeat_ngram_size": 0,
46
+ "eos_token_id": null,
47
+ "exponential_decay_length_penalty": null,
48
+ "finetuning_task": null,
49
+ "forced_bos_token_id": null,
50
+ "forced_eos_token_id": null,
51
+ "hidden_act": "gelu",
52
+ "hidden_dropout_prob": 0.0,
53
+ "hidden_sizes": [
54
+ 32,
55
+ 64,
56
+ 160,
57
+ 256
58
+ ],
59
+ "id2label": {
60
+ "0": "unlabeled",
61
+ "1": "building",
62
+ "2": "fence",
63
+ "3": "other",
64
+ "4": "pedestrian",
65
+ "5": "pole",
66
+ "6": "road line",
67
+ "7": "road",
68
+ "8": "sidewalk",
69
+ "9": "vegetation",
70
+ "10": "vehicle",
71
+ "11": "wall",
72
+ "12": "traffic sign",
73
+ "13": "sky",
74
+ "14": "ground",
75
+ "15": "bridge",
76
+ "16": "rail track",
77
+ "17": "guard rail",
78
+ "18": "traffic light",
79
+ "19": "static",
80
+ "20": "dynamic",
81
+ "21": "water",
82
+ "22": "terrain"
83
+ },
84
+ "image_size": {
85
+ "height": 224,
86
+ "width": 224
87
+ },
88
+ "initializer_range": 0.02,
89
+ "is_decoder": false,
90
+ "is_encoder_decoder": false,
91
+ "label2id": {
92
+ "bridge": 15,
93
+ "building": 1,
94
+ "dynamic": 20,
95
+ "fence": 2,
96
+ "ground": 14,
97
+ "guard rail": 17,
98
+ "other": 3,
99
+ "pedestrian": 4,
100
+ "pole": 5,
101
+ "rail track": 16,
102
+ "road": 7,
103
+ "road line": 6,
104
+ "sidewalk": 8,
105
+ "sky": 13,
106
+ "static": 19,
107
+ "terrain": 22,
108
+ "traffic light": 18,
109
+ "traffic sign": 12,
110
+ "unlabeled": 0,
111
+ "vegetation": 9,
112
+ "vehicle": 10,
113
+ "wall": 11,
114
+ "water": 21
115
+ },
116
+ "layer_norm_eps": 1e-06,
117
+ "length_penalty": 1.0,
118
+ "max_length": 20,
119
+ "min_length": 0,
120
+ "mlp_ratios": [
121
+ 4,
122
+ 4,
123
+ 4,
124
+ 4
125
+ ],
126
+ "model_type": "pvt_v2",
127
+ "no_repeat_ngram_size": 0,
128
+ "num_attention_heads": [
129
+ 1,
130
+ 2,
131
+ 5,
132
+ 8
133
+ ],
134
+ "num_beam_groups": 1,
135
+ "num_beams": 1,
136
+ "num_channels": 3,
137
+ "num_encoder_blocks": 4,
138
+ "num_return_sequences": 1,
139
+ "output_attentions": false,
140
+ "output_hidden_states": true,
141
+ "output_scores": false,
142
+ "pad_token_id": null,
143
+ "patch_sizes": [
144
+ 7,
145
+ 3,
146
+ 3,
147
+ 3
148
+ ],
149
+ "prefix": null,
150
+ "problem_type": null,
151
+ "pruned_heads": {},
152
+ "qkv_bias": true,
153
+ "remove_invalid_values": false,
154
+ "repetition_penalty": 1.0,
155
+ "reshape_last_stage": true,
156
+ "return_dict": true,
157
+ "return_dict_in_generate": false,
158
+ "sep_token_id": null,
159
+ "sr_ratios": [
160
+ 8,
161
+ 4,
162
+ 2,
163
+ 1
164
+ ],
165
+ "stage_names": [
166
+ "stem",
167
+ "stage1",
168
+ "stage2",
169
+ "stage3",
170
+ "stage4"
171
+ ],
172
+ "strides": [
173
+ 4,
174
+ 2,
175
+ 2,
176
+ 2
177
+ ],
178
+ "suppress_tokens": null,
179
+ "task_specific_params": null,
180
+ "temperature": 1.0,
181
+ "tf_legacy_loss": false,
182
+ "tie_encoder_decoder": false,
183
+ "tie_word_embeddings": true,
184
+ "tokenizer_class": null,
185
+ "top_k": 50,
186
+ "top_p": 1.0,
187
+ "torch_dtype": null,
188
+ "torchscript": false,
189
+ "typical_p": 1.0,
190
+ "use_bfloat16": false
191
+ },
192
+ "bbox_cost": 5,
193
+ "bbox_loss_coefficient": 5,
194
+ "class_cost": 1,
195
+ "d_model": 256,
196
+ "decoder_attention_heads": 8,
197
+ "decoder_ffn_dim": 256,
198
+ "decoder_layers": 3,
199
+ "decoder_n_points": 4,
200
+ "depth_decoder_dim": 64,
201
+ "dice_loss_coefficient": 1,
202
+ "dilation": false,
203
+ "disable_custom_kernels": false,
204
+ "dropout": 0.1,
205
+ "encoder_attention_heads": 8,
206
+ "encoder_ffn_dim": 256,
207
+ "encoder_layerdrop": 0.0,
208
+ "encoder_layers": 3,
209
+ "encoder_n_points": 4,
210
+ "eos_coefficient": 0.1,
211
+ "focal_alpha": 0.25,
212
+ "frozen_batch_norm": false,
213
+ "giou_cost": 2,
214
+ "giou_loss_coefficient": 2,
215
+ "head_in_index": -1,
216
+ "id2label": {
217
+ "0": "pedestrian",
218
+ "1": "car",
219
+ "2": "truck",
220
+ "3": "bus",
221
+ "4": "motorcycle",
222
+ "5": "bicycle",
223
+ "6": "no_box"
224
+ },
225
+ "init_std": 0.02,
226
+ "init_xavier_std": 1.0,
227
+ "is_encoder_decoder": true,
228
+ "label2id": null,
229
+ "mask_loss_coefficient": 1,
230
+ "max_depth": 10,
231
+ "max_position_embeddings": 1024,
232
+ "model_type": "multiformer",
233
+ "num_channels": 3,
234
+ "num_feature_levels": 4,
235
+ "num_queries": 300,
236
+ "position_embedding_type": "sine",
237
+ "semantic_classifier_dropout": 0.1,
238
+ "semantic_decoder_dim": 256,
239
+ "semantic_loss_ignore_index": 255,
240
+ "silog_lambda": 0.25,
241
+ "torch_dtype": "float32",
242
+ "transformers_version": "4.33.0.dev0",
243
+ "two_stage": false,
244
+ "two_stage_num_proposals": 300,
245
+ "use_pretrained_backbone": true,
246
+ "use_timm_backbone": false,
247
+ "with_box_refine": false
248
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3b2cda0bbe276c5f835ae40c74efa00b13c3014b8680ad34fc982b236d9b8872
3
+ size 32310749