zarqankhn commited on
Commit
37e224d
·
verified ·
1 Parent(s): e547acf

Upload model_metadata.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. model_metadata.json +408 -0
model_metadata.json ADDED
@@ -0,0 +1,408 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "quickdraw_345_se_resnet",
3
+ "description": "Quick Draw doodle classifier (345 categories) for Flutter on-device inference",
4
+ "version": "1.0.0",
5
+ "num_classes": 345,
6
+ "image_size": 28,
7
+ "channels": 1,
8
+ "color_mode": "grayscale",
9
+ "input_shape": [
10
+ 1,
11
+ 28,
12
+ 28,
13
+ 1
14
+ ],
15
+ "output_shape": [
16
+ 1,
17
+ 345
18
+ ],
19
+ "input_dtype": "float32",
20
+ "normalization": "divide_by_255",
21
+ "background_value": "1.0 (white)",
22
+ "stroke_value": "0.0 (black)",
23
+ "accuracy": {
24
+ "top1": 0.7619,
25
+ "top3": 0.8951,
26
+ "top5": 0.9226,
27
+ "top10": 0.9455,
28
+ "tflite_f16": 0.764
29
+ },
30
+ "model_files": {
31
+ "float16": {
32
+ "filename": "quickdraw_model.tflite",
33
+ "size_mb": 8.44
34
+ },
35
+ "int8": {
36
+ "filename": "quickdraw_model_int8.tflite",
37
+ "size_mb": 4.34
38
+ }
39
+ },
40
+ "training": {
41
+ "architecture": "SE-ResNet (64->128->256)",
42
+ "dataset": "Google Quick Draw numpy bitmaps",
43
+ "samples_per_class": 8000,
44
+ "total_samples": 2760000,
45
+ "train_samples": 2346000,
46
+ "val_samples": 414000,
47
+ "epochs_trained": 60,
48
+ "batch_size": 1024,
49
+ "optimizer": "Adam + WarmupCosineDecay",
50
+ "augmentation": [
51
+ "rotation(+-8%)",
52
+ "translation(+-8%)",
53
+ "zoom(-5%,+10%)"
54
+ ],
55
+ "training_time_minutes": 657.3
56
+ },
57
+ "flutter_integration": {
58
+ "package": "tflite_flutter: ^0.10.4",
59
+ "usage": "See model_metadata.json for Flutter integration example"
60
+ },
61
+ "categories": [
62
+ "The Eiffel Tower",
63
+ "The Great Wall of China",
64
+ "The Mona Lisa",
65
+ "aircraft carrier",
66
+ "airplane",
67
+ "alarm clock",
68
+ "ambulance",
69
+ "angel",
70
+ "animal migration",
71
+ "ant",
72
+ "anvil",
73
+ "apple",
74
+ "arm",
75
+ "asparagus",
76
+ "axe",
77
+ "backpack",
78
+ "banana",
79
+ "bandage",
80
+ "barn",
81
+ "baseball",
82
+ "baseball bat",
83
+ "basket",
84
+ "basketball",
85
+ "bat",
86
+ "bathtub",
87
+ "beach",
88
+ "bear",
89
+ "beard",
90
+ "bed",
91
+ "bee",
92
+ "belt",
93
+ "bench",
94
+ "bicycle",
95
+ "binoculars",
96
+ "bird",
97
+ "birthday cake",
98
+ "blackberry",
99
+ "blueberry",
100
+ "book",
101
+ "boomerang",
102
+ "bottlecap",
103
+ "bowtie",
104
+ "bracelet",
105
+ "brain",
106
+ "bread",
107
+ "bridge",
108
+ "broccoli",
109
+ "broom",
110
+ "bucket",
111
+ "bulldozer",
112
+ "bus",
113
+ "bush",
114
+ "butterfly",
115
+ "cactus",
116
+ "cake",
117
+ "calculator",
118
+ "calendar",
119
+ "camel",
120
+ "camera",
121
+ "camouflage",
122
+ "campfire",
123
+ "candle",
124
+ "cannon",
125
+ "canoe",
126
+ "car",
127
+ "carrot",
128
+ "castle",
129
+ "cat",
130
+ "ceiling fan",
131
+ "cell phone",
132
+ "cello",
133
+ "chair",
134
+ "chandelier",
135
+ "church",
136
+ "circle",
137
+ "clarinet",
138
+ "clock",
139
+ "cloud",
140
+ "coffee cup",
141
+ "compass",
142
+ "computer",
143
+ "cookie",
144
+ "cooler",
145
+ "couch",
146
+ "cow",
147
+ "crab",
148
+ "crayon",
149
+ "crocodile",
150
+ "crown",
151
+ "cruise ship",
152
+ "cup",
153
+ "diamond",
154
+ "dishwasher",
155
+ "diving board",
156
+ "dog",
157
+ "dolphin",
158
+ "donut",
159
+ "door",
160
+ "dragon",
161
+ "dresser",
162
+ "drill",
163
+ "drums",
164
+ "duck",
165
+ "dumbbell",
166
+ "ear",
167
+ "elbow",
168
+ "elephant",
169
+ "envelope",
170
+ "eraser",
171
+ "eye",
172
+ "eyeglasses",
173
+ "face",
174
+ "fan",
175
+ "feather",
176
+ "fence",
177
+ "finger",
178
+ "fire hydrant",
179
+ "fireplace",
180
+ "firetruck",
181
+ "fish",
182
+ "flamingo",
183
+ "flashlight",
184
+ "flip flops",
185
+ "floor lamp",
186
+ "flower",
187
+ "flying saucer",
188
+ "foot",
189
+ "fork",
190
+ "frog",
191
+ "frying pan",
192
+ "garden",
193
+ "garden hose",
194
+ "giraffe",
195
+ "goatee",
196
+ "golf club",
197
+ "grapes",
198
+ "grass",
199
+ "guitar",
200
+ "hamburger",
201
+ "hammer",
202
+ "hand",
203
+ "harp",
204
+ "hat",
205
+ "headphones",
206
+ "hedgehog",
207
+ "helicopter",
208
+ "helmet",
209
+ "hexagon",
210
+ "hockey puck",
211
+ "hockey stick",
212
+ "horse",
213
+ "hospital",
214
+ "hot air balloon",
215
+ "hot dog",
216
+ "hot tub",
217
+ "hourglass",
218
+ "house",
219
+ "house plant",
220
+ "hurricane",
221
+ "ice cream",
222
+ "jacket",
223
+ "jail",
224
+ "kangaroo",
225
+ "key",
226
+ "keyboard",
227
+ "knee",
228
+ "knife",
229
+ "ladder",
230
+ "lantern",
231
+ "laptop",
232
+ "leaf",
233
+ "leg",
234
+ "light bulb",
235
+ "lighter",
236
+ "lighthouse",
237
+ "lightning",
238
+ "line",
239
+ "lion",
240
+ "lipstick",
241
+ "lobster",
242
+ "lollipop",
243
+ "mailbox",
244
+ "map",
245
+ "marker",
246
+ "matches",
247
+ "megaphone",
248
+ "mermaid",
249
+ "microphone",
250
+ "microwave",
251
+ "monkey",
252
+ "moon",
253
+ "mosquito",
254
+ "motorbike",
255
+ "mountain",
256
+ "mouse",
257
+ "moustache",
258
+ "mouth",
259
+ "mug",
260
+ "mushroom",
261
+ "nail",
262
+ "necklace",
263
+ "nose",
264
+ "ocean",
265
+ "octagon",
266
+ "octopus",
267
+ "onion",
268
+ "oven",
269
+ "owl",
270
+ "paint can",
271
+ "paintbrush",
272
+ "palm tree",
273
+ "panda",
274
+ "pants",
275
+ "paper clip",
276
+ "parachute",
277
+ "parrot",
278
+ "passport",
279
+ "peanut",
280
+ "pear",
281
+ "peas",
282
+ "pencil",
283
+ "penguin",
284
+ "piano",
285
+ "pickup truck",
286
+ "picture frame",
287
+ "pig",
288
+ "pillow",
289
+ "pineapple",
290
+ "pizza",
291
+ "pliers",
292
+ "police car",
293
+ "pond",
294
+ "pool",
295
+ "popsicle",
296
+ "postcard",
297
+ "potato",
298
+ "power outlet",
299
+ "purse",
300
+ "rabbit",
301
+ "raccoon",
302
+ "radio",
303
+ "rain",
304
+ "rainbow",
305
+ "rake",
306
+ "remote control",
307
+ "rhinoceros",
308
+ "rifle",
309
+ "river",
310
+ "roller coaster",
311
+ "rollerskates",
312
+ "sailboat",
313
+ "sandwich",
314
+ "saw",
315
+ "saxophone",
316
+ "school bus",
317
+ "scissors",
318
+ "scorpion",
319
+ "screwdriver",
320
+ "sea turtle",
321
+ "see saw",
322
+ "shark",
323
+ "sheep",
324
+ "shoe",
325
+ "shorts",
326
+ "shovel",
327
+ "sink",
328
+ "skateboard",
329
+ "skull",
330
+ "skyscraper",
331
+ "sleeping bag",
332
+ "smiley face",
333
+ "snail",
334
+ "snake",
335
+ "snorkel",
336
+ "snowflake",
337
+ "snowman",
338
+ "soccer ball",
339
+ "sock",
340
+ "speedboat",
341
+ "spider",
342
+ "spoon",
343
+ "spreadsheet",
344
+ "square",
345
+ "squiggle",
346
+ "squirrel",
347
+ "stairs",
348
+ "star",
349
+ "steak",
350
+ "stereo",
351
+ "stethoscope",
352
+ "stitches",
353
+ "stop sign",
354
+ "stove",
355
+ "strawberry",
356
+ "streetlight",
357
+ "string bean",
358
+ "submarine",
359
+ "suitcase",
360
+ "sun",
361
+ "swan",
362
+ "sweater",
363
+ "swing set",
364
+ "sword",
365
+ "syringe",
366
+ "t-shirt",
367
+ "table",
368
+ "teapot",
369
+ "teddy-bear",
370
+ "telephone",
371
+ "television",
372
+ "tennis racquet",
373
+ "tent",
374
+ "tiger",
375
+ "toaster",
376
+ "toe",
377
+ "toilet",
378
+ "tooth",
379
+ "toothbrush",
380
+ "toothpaste",
381
+ "tornado",
382
+ "tractor",
383
+ "traffic light",
384
+ "train",
385
+ "tree",
386
+ "triangle",
387
+ "trombone",
388
+ "truck",
389
+ "trumpet",
390
+ "umbrella",
391
+ "underwear",
392
+ "van",
393
+ "vase",
394
+ "violin",
395
+ "washing machine",
396
+ "watermelon",
397
+ "waterslide",
398
+ "whale",
399
+ "wheel",
400
+ "windmill",
401
+ "wine bottle",
402
+ "wine glass",
403
+ "wristwatch",
404
+ "yoga",
405
+ "zebra",
406
+ "zigzag"
407
+ ]
408
+ }