sbchoi commited on
Commit
b4f061c
1 Parent(s): 60b1487

Upload Yolov6ForObjectDetection

Browse files
Files changed (2) hide show
  1. config.json +238 -0
  2. model.safetensors +3 -0
config.json ADDED
@@ -0,0 +1,238 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "Yolov6ForObjectDetection"
4
+ ],
5
+ "atss_warmup_epoch": 0,
6
+ "backbone_csp_e": 0.5,
7
+ "backbone_cspsppf": false,
8
+ "backbone_fuse_P2": true,
9
+ "backbone_num_repeats": [
10
+ 1,
11
+ 6,
12
+ 12,
13
+ 18,
14
+ 6,
15
+ 6
16
+ ],
17
+ "backbone_out_channels": [
18
+ 64,
19
+ 128,
20
+ 256,
21
+ 512,
22
+ 768,
23
+ 1024
24
+ ],
25
+ "backbone_stage_block_type": "Yolov6BepC3",
26
+ "block_type": "Yolov6ConvBNSilu",
27
+ "class_loss_coefficient": 1.0,
28
+ "dfl_loss_coefficient": 1.0,
29
+ "export": false,
30
+ "forward_fuse": false,
31
+ "head_anchors": 1,
32
+ "head_in_channels": [
33
+ 128,
34
+ 256,
35
+ 512,
36
+ 1024
37
+ ],
38
+ "head_num_layers": 4,
39
+ "head_strides": [
40
+ 8,
41
+ 16,
42
+ 32,
43
+ 64
44
+ ],
45
+ "id2label": {
46
+ "0": "person",
47
+ "1": "bicycle",
48
+ "2": "car",
49
+ "3": "motorbike",
50
+ "4": "aeroplane",
51
+ "5": "bus",
52
+ "6": "train",
53
+ "7": "truck",
54
+ "8": "boat",
55
+ "9": "traffic light",
56
+ "10": "fire hydrant",
57
+ "11": "stop sign",
58
+ "12": "parking meter",
59
+ "13": "bench",
60
+ "14": "bird",
61
+ "15": "cat",
62
+ "16": "dog",
63
+ "17": "horse",
64
+ "18": "sheep",
65
+ "19": "cow",
66
+ "20": "elephant",
67
+ "21": "bear",
68
+ "22": "zebra",
69
+ "23": "giraffe",
70
+ "24": "backpack",
71
+ "25": "umbrella",
72
+ "26": "handbag",
73
+ "27": "tie",
74
+ "28": "suitcase",
75
+ "29": "frisbee",
76
+ "30": "skis",
77
+ "31": "snowboard",
78
+ "32": "sports ball",
79
+ "33": "kite",
80
+ "34": "baseball bat",
81
+ "35": "baseball glove",
82
+ "36": "skateboard",
83
+ "37": "surfboard",
84
+ "38": "tennis racket",
85
+ "39": "bottle",
86
+ "40": "wine glass",
87
+ "41": "cup",
88
+ "42": "fork",
89
+ "43": "knife",
90
+ "44": "spoon",
91
+ "45": "bowl",
92
+ "46": "banana",
93
+ "47": "apple",
94
+ "48": "sandwich",
95
+ "49": "orange",
96
+ "50": "broccoli",
97
+ "51": "carrot",
98
+ "52": "hot dog",
99
+ "53": "pizza",
100
+ "54": "donut",
101
+ "55": "cake",
102
+ "56": "chair",
103
+ "57": "sofa",
104
+ "58": "pottedplant",
105
+ "59": "bed",
106
+ "60": "diningtable",
107
+ "61": "toilet",
108
+ "62": "tvmonitor",
109
+ "63": "laptop",
110
+ "64": "mouse",
111
+ "65": "remote",
112
+ "66": "keyboard",
113
+ "67": "cell phone",
114
+ "68": "microwave",
115
+ "69": "oven",
116
+ "70": "toaster",
117
+ "71": "sink",
118
+ "72": "refrigerator",
119
+ "73": "book",
120
+ "74": "clock",
121
+ "75": "vase",
122
+ "76": "scissors",
123
+ "77": "teddy bear",
124
+ "78": "hair drier",
125
+ "79": "toothbrush"
126
+ },
127
+ "image_size": 1280,
128
+ "in_channels": 3,
129
+ "initializer_range": 0.02,
130
+ "iou_loss_coefficient": 1.0,
131
+ "iou_type": "giou",
132
+ "label2id": {
133
+ "aeroplane": 4,
134
+ "apple": 47,
135
+ "backpack": 24,
136
+ "banana": 46,
137
+ "baseball bat": 34,
138
+ "baseball glove": 35,
139
+ "bear": 21,
140
+ "bed": 59,
141
+ "bench": 13,
142
+ "bicycle": 1,
143
+ "bird": 14,
144
+ "boat": 8,
145
+ "book": 73,
146
+ "bottle": 39,
147
+ "bowl": 45,
148
+ "broccoli": 50,
149
+ "bus": 5,
150
+ "cake": 55,
151
+ "car": 2,
152
+ "carrot": 51,
153
+ "cat": 15,
154
+ "cell phone": 67,
155
+ "chair": 56,
156
+ "clock": 74,
157
+ "cow": 19,
158
+ "cup": 41,
159
+ "diningtable": 60,
160
+ "dog": 16,
161
+ "donut": 54,
162
+ "elephant": 20,
163
+ "fire hydrant": 10,
164
+ "fork": 42,
165
+ "frisbee": 29,
166
+ "giraffe": 23,
167
+ "hair drier": 78,
168
+ "handbag": 26,
169
+ "horse": 17,
170
+ "hot dog": 52,
171
+ "keyboard": 66,
172
+ "kite": 33,
173
+ "knife": 43,
174
+ "laptop": 63,
175
+ "microwave": 68,
176
+ "motorbike": 3,
177
+ "mouse": 64,
178
+ "orange": 49,
179
+ "oven": 69,
180
+ "parking meter": 12,
181
+ "person": 0,
182
+ "pizza": 53,
183
+ "pottedplant": 58,
184
+ "refrigerator": 72,
185
+ "remote": 65,
186
+ "sandwich": 48,
187
+ "scissors": 76,
188
+ "sheep": 18,
189
+ "sink": 71,
190
+ "skateboard": 36,
191
+ "skis": 30,
192
+ "snowboard": 31,
193
+ "sofa": 57,
194
+ "spoon": 44,
195
+ "sports ball": 32,
196
+ "stop sign": 11,
197
+ "suitcase": 28,
198
+ "surfboard": 37,
199
+ "teddy bear": 77,
200
+ "tennis racket": 38,
201
+ "tie": 27,
202
+ "toaster": 70,
203
+ "toilet": 61,
204
+ "toothbrush": 79,
205
+ "traffic light": 9,
206
+ "train": 6,
207
+ "truck": 7,
208
+ "tvmonitor": 62,
209
+ "umbrella": 25,
210
+ "vase": 75,
211
+ "wine glass": 40,
212
+ "zebra": 22
213
+ },
214
+ "model_type": "yolov6n",
215
+ "neck_csp_e": 0.5,
216
+ "neck_num_repeats": [
217
+ 12,
218
+ 12,
219
+ 12,
220
+ 12,
221
+ 12,
222
+ 12
223
+ ],
224
+ "neck_out_channels": [
225
+ 512,
226
+ 256,
227
+ 128,
228
+ 256,
229
+ 512,
230
+ 1024
231
+ ],
232
+ "neck_stage_block_type": "Yolov6BepC3",
233
+ "reg_max": 16,
234
+ "reg_max_proj": 16,
235
+ "torch_dtype": "float32",
236
+ "transformers_version": "4.38.0.dev0",
237
+ "use_dfl": true
238
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e3e8cb365b4127b6b1a22cfe15bc9c7b5b14070154062a79754ff7286cf4dfba
3
+ size 562247904