config.json
4.5 KB · 227 lines · json Raw
1 {
2 "activation_dropout": 0.0,
3 "activation_function": "relu",
4 "architectures": [
5 "DetrForObjectDetection"
6 ],
7 "attention_dropout": 0.0,
8 "auxiliary_loss": false,
9 "backbone": "resnet50",
10 "bbox_cost": 5,
11 "bbox_loss_coefficient": 5,
12 "class_cost": 1,
13 "classifier_dropout": 0.0,
14 "d_model": 256,
15 "decoder_attention_heads": 8,
16 "decoder_ffn_dim": 2048,
17 "decoder_layerdrop": 0.0,
18 "decoder_layers": 6,
19 "dice_loss_coefficient": 1,
20 "dilation": false,
21 "dropout": 0.1,
22 "encoder_attention_heads": 8,
23 "encoder_ffn_dim": 2048,
24 "encoder_layerdrop": 0.0,
25 "encoder_layers": 6,
26 "eos_coefficient": 0.1,
27 "giou_cost": 2,
28 "giou_loss_coefficient": 2,
29 "id2label": {
30 "0": "N/A",
31 "1": "person",
32 "10": "traffic light",
33 "11": "fire hydrant",
34 "12": "street sign",
35 "13": "stop sign",
36 "14": "parking meter",
37 "15": "bench",
38 "16": "bird",
39 "17": "cat",
40 "18": "dog",
41 "19": "horse",
42 "2": "bicycle",
43 "20": "sheep",
44 "21": "cow",
45 "22": "elephant",
46 "23": "bear",
47 "24": "zebra",
48 "25": "giraffe",
49 "26": "hat",
50 "27": "backpack",
51 "28": "umbrella",
52 "29": "shoe",
53 "3": "car",
54 "30": "eye glasses",
55 "31": "handbag",
56 "32": "tie",
57 "33": "suitcase",
58 "34": "frisbee",
59 "35": "skis",
60 "36": "snowboard",
61 "37": "sports ball",
62 "38": "kite",
63 "39": "baseball bat",
64 "4": "motorcycle",
65 "40": "baseball glove",
66 "41": "skateboard",
67 "42": "surfboard",
68 "43": "tennis racket",
69 "44": "bottle",
70 "45": "plate",
71 "46": "wine glass",
72 "47": "cup",
73 "48": "fork",
74 "49": "knife",
75 "5": "airplane",
76 "50": "spoon",
77 "51": "bowl",
78 "52": "banana",
79 "53": "apple",
80 "54": "sandwich",
81 "55": "orange",
82 "56": "broccoli",
83 "57": "carrot",
84 "58": "hot dog",
85 "59": "pizza",
86 "6": "bus",
87 "60": "donut",
88 "61": "cake",
89 "62": "chair",
90 "63": "couch",
91 "64": "potted plant",
92 "65": "bed",
93 "66": "mirror",
94 "67": "dining table",
95 "68": "window",
96 "69": "desk",
97 "7": "train",
98 "70": "toilet",
99 "71": "door",
100 "72": "tv",
101 "73": "laptop",
102 "74": "mouse",
103 "75": "remote",
104 "76": "keyboard",
105 "77": "cell phone",
106 "78": "microwave",
107 "79": "oven",
108 "8": "truck",
109 "80": "toaster",
110 "81": "sink",
111 "82": "refrigerator",
112 "83": "blender",
113 "84": "book",
114 "85": "clock",
115 "86": "vase",
116 "87": "scissors",
117 "88": "teddy bear",
118 "89": "hair drier",
119 "9": "boat",
120 "90": "toothbrush"
121 },
122 "init_std": 0.02,
123 "init_xavier_std": 1.0,
124 "is_encoder_decoder": true,
125 "label2id": {
126 "N/A": 0,
127 "airplane": 5,
128 "apple": 53,
129 "backpack": 27,
130 "banana": 52,
131 "baseball bat": 39,
132 "baseball glove": 40,
133 "bear": 23,
134 "bed": 65,
135 "bench": 15,
136 "bicycle": 2,
137 "bird": 16,
138 "blender": 83,
139 "boat": 9,
140 "book": 84,
141 "bottle": 44,
142 "bowl": 51,
143 "broccoli": 56,
144 "bus": 6,
145 "cake": 61,
146 "car": 3,
147 "carrot": 57,
148 "cat": 17,
149 "cell phone": 77,
150 "chair": 62,
151 "clock": 85,
152 "couch": 63,
153 "cow": 21,
154 "cup": 47,
155 "desk": 69,
156 "dining table": 67,
157 "dog": 18,
158 "donut": 60,
159 "door": 71,
160 "elephant": 22,
161 "eye glasses": 30,
162 "fire hydrant": 11,
163 "fork": 48,
164 "frisbee": 34,
165 "giraffe": 25,
166 "hair drier": 89,
167 "handbag": 31,
168 "hat": 26,
169 "horse": 19,
170 "hot dog": 58,
171 "keyboard": 76,
172 "kite": 38,
173 "knife": 49,
174 "laptop": 73,
175 "microwave": 78,
176 "mirror": 66,
177 "motorcycle": 4,
178 "mouse": 74,
179 "orange": 55,
180 "oven": 79,
181 "parking meter": 14,
182 "person": 1,
183 "pizza": 59,
184 "plate": 45,
185 "potted plant": 64,
186 "refrigerator": 82,
187 "remote": 75,
188 "sandwich": 54,
189 "scissors": 87,
190 "sheep": 20,
191 "shoe": 29,
192 "sink": 81,
193 "skateboard": 41,
194 "skis": 35,
195 "snowboard": 36,
196 "spoon": 50,
197 "sports ball": 37,
198 "stop sign": 13,
199 "street sign": 12,
200 "suitcase": 33,
201 "surfboard": 42,
202 "teddy bear": 88,
203 "tennis racket": 43,
204 "tie": 32,
205 "toaster": 80,
206 "toilet": 70,
207 "toothbrush": 90,
208 "traffic light": 10,
209 "train": 7,
210 "truck": 8,
211 "tv": 72,
212 "umbrella": 28,
213 "vase": 86,
214 "window": 68,
215 "wine glass": 46,
216 "zebra": 24
217 },
218 "mask_loss_coefficient": 1,
219 "max_position_embeddings": 1024,
220 "model_type": "detr",
221 "num_hidden_layers": 6,
222 "num_queries": 100,
223 "position_embedding_type": "sine",
224 "scale_embedding": false,
225 "transformers_version": "4.7.0.dev0"
226 }
227