nehulagrawal commited on
Commit
7aeb5b6
1 Parent(s): c19b9f6

Create config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -209
config.json CHANGED
@@ -1,209 +1 @@
1
- {
2
- "architectures": [
3
- "YolosForObjectDetection"
4
- ],
5
- "attention_probs_dropout_prob": 0.0,
6
- "auxiliary_loss": false,
7
- "bbox_cost": 5,
8
- "bbox_loss_coefficient": 5,
9
- "class_cost": 1,
10
- "eos_coefficient": 0.1,
11
- "giou_cost": 2,
12
- "giou_loss_coefficient": 2,
13
- "hidden_act": "gelu",
14
- "hidden_dropout_prob": 0.0,
15
- "hidden_size": 192,
16
- "id2label": {
17
- "0": "N/A",
18
- "1": "person",
19
- "2": "bicycle",
20
- "3": "car",
21
- "4": "motorcycle",
22
- "5": "airplane",
23
- "6": "bus",
24
- "7": "train",
25
- "8": "truck",
26
- "9": "boat",
27
- "10": "traffic light",
28
- "11": "fire hydrant",
29
- "12": "N/A",
30
- "13": "stop sign",
31
- "14": "parking meter",
32
- "15": "bench",
33
- "16": "bird",
34
- "17": "cat",
35
- "18": "dog",
36
- "19": "horse",
37
- "20": "sheep",
38
- "21": "cow",
39
- "22": "elephant",
40
- "23": "bear",
41
- "24": "zebra",
42
- "25": "giraffe",
43
- "26": "N/A",
44
- "27": "backpack",
45
- "28": "umbrella",
46
- "29": "N/A",
47
- "30": "N/A",
48
- "31": "handbag",
49
- "32": "tie",
50
- "33": "suitcase",
51
- "34": "frisbee",
52
- "35": "skis",
53
- "36": "snowboard",
54
- "37": "sports ball",
55
- "38": "kite",
56
- "39": "baseball bat",
57
- "40": "baseball glove",
58
- "41": "skateboard",
59
- "42": "surfboard",
60
- "43": "tennis racket",
61
- "44": "bottle",
62
- "45": "N/A",
63
- "46": "wine glass",
64
- "47": "cup",
65
- "48": "fork",
66
- "49": "knife",
67
- "50": "spoon",
68
- "51": "bowl",
69
- "52": "banana",
70
- "53": "apple",
71
- "54": "sandwich",
72
- "55": "orange",
73
- "56": "broccoli",
74
- "57": "carrot",
75
- "58": "hot dog",
76
- "59": "pizza",
77
- "60": "donut",
78
- "61": "cake",
79
- "62": "chair",
80
- "63": "couch",
81
- "64": "potted plant",
82
- "65": "bed",
83
- "66": "N/A",
84
- "67": "dining table",
85
- "68": "N/A",
86
- "69": "N/A",
87
- "70": "toilet",
88
- "71": "N/A",
89
- "72": "tv",
90
- "73": "laptop",
91
- "74": "mouse",
92
- "75": "remote",
93
- "76": "keyboard",
94
- "77": "cell phone",
95
- "78": "microwave",
96
- "79": "oven",
97
- "80": "toaster",
98
- "81": "sink",
99
- "82": "refrigerator",
100
- "83": "N/A",
101
- "84": "book",
102
- "85": "clock",
103
- "86": "vase",
104
- "87": "scissors",
105
- "88": "teddy bear",
106
- "89": "hair drier",
107
- "90": "toothbrush"
108
- },
109
- "image_size": [
110
- 800,
111
- 1333
112
- ],
113
- "initializer_range": 0.02,
114
- "intermediate_size": 768,
115
- "label2id": {
116
- "N/A": 83,
117
- "airplane": 5,
118
- "apple": 53,
119
- "backpack": 27,
120
- "banana": 52,
121
- "baseball bat": 39,
122
- "baseball glove": 40,
123
- "bear": 23,
124
- "bed": 65,
125
- "bench": 15,
126
- "bicycle": 2,
127
- "bird": 16,
128
- "boat": 9,
129
- "book": 84,
130
- "bottle": 44,
131
- "bowl": 51,
132
- "broccoli": 56,
133
- "bus": 6,
134
- "cake": 61,
135
- "car": 3,
136
- "carrot": 57,
137
- "cat": 17,
138
- "cell phone": 77,
139
- "chair": 62,
140
- "clock": 85,
141
- "couch": 63,
142
- "cow": 21,
143
- "cup": 47,
144
- "dining table": 67,
145
- "dog": 18,
146
- "donut": 60,
147
- "elephant": 22,
148
- "fire hydrant": 11,
149
- "fork": 48,
150
- "frisbee": 34,
151
- "giraffe": 25,
152
- "hair drier": 89,
153
- "handbag": 31,
154
- "horse": 19,
155
- "hot dog": 58,
156
- "keyboard": 76,
157
- "kite": 38,
158
- "knife": 49,
159
- "laptop": 73,
160
- "microwave": 78,
161
- "motorcycle": 4,
162
- "mouse": 74,
163
- "orange": 55,
164
- "oven": 79,
165
- "parking meter": 14,
166
- "person": 1,
167
- "pizza": 59,
168
- "potted plant": 64,
169
- "refrigerator": 82,
170
- "remote": 75,
171
- "sandwich": 54,
172
- "scissors": 87,
173
- "sheep": 20,
174
- "sink": 81,
175
- "skateboard": 41,
176
- "skis": 35,
177
- "snowboard": 36,
178
- "spoon": 50,
179
- "sports ball": 37,
180
- "stop sign": 13,
181
- "suitcase": 33,
182
- "surfboard": 42,
183
- "teddy bear": 88,
184
- "tennis racket": 43,
185
- "tie": 32,
186
- "toaster": 80,
187
- "toilet": 70,
188
- "toothbrush": 90,
189
- "traffic light": 10,
190
- "train": 7,
191
- "truck": 8,
192
- "tv": 72,
193
- "umbrella": 28,
194
- "vase": 86,
195
- "wine glass": 46,
196
- "zebra": 24
197
- },
198
- "layer_norm_eps": 1e-12,
199
- "model_type": "yolos",
200
- "num_attention_heads": 3,
201
- "num_channels": 3,
202
- "num_detection_tokens": 100,
203
- "num_hidden_layers": 12,
204
- "patch_size": 16,
205
- "qkv_bias": true,
206
- "torch_dtype": "float32",
207
- "transformers_version": "4.19.0.dev0",
208
- "use_mid_position_embeddings": false
209
- }
 
1
+ {"input_size": 640, "task": "object-detection", "ultralyticsplus_version": "0.0.28", "ultralytics_version": "8.0.43", "model_type": "v8", "score_map50": 0.95196}