devarajns commited on
Commit
31ab753
·
verified ·
1 Parent(s): 2158981

Upload ONNX models

Browse files
models/.DS_Store ADDED
Binary file (6.15 kB). View file
 
models/quickdraw-mvp/model.json ADDED
@@ -0,0 +1,800 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schemaVersion": 1,
3
+ "kind": "quickdraw-classifier",
4
+ "classes": [
5
+ "airplane",
6
+ "apple",
7
+ "banana",
8
+ "bicycle",
9
+ "bird",
10
+ "book",
11
+ "bus",
12
+ "cake",
13
+ "camera",
14
+ "car",
15
+ "cat",
16
+ "chair",
17
+ "clock",
18
+ "cloud",
19
+ "coffee cup",
20
+ "computer",
21
+ "dog",
22
+ "door",
23
+ "eye",
24
+ "face",
25
+ "fish",
26
+ "flower",
27
+ "hamburger",
28
+ "hand",
29
+ "house",
30
+ "ice cream",
31
+ "key",
32
+ "light bulb",
33
+ "moon",
34
+ "mountain",
35
+ "pencil",
36
+ "pizza",
37
+ "rain",
38
+ "rainbow",
39
+ "star",
40
+ "sun",
41
+ "telephone",
42
+ "tree",
43
+ "truck",
44
+ "umbrella"
45
+ ],
46
+ "sourceCacheFingerprint": "5d56d8dd9ccc445000f843298d921611e777f90010f75bdfd344988a67b2f4bb",
47
+ "trainingData": {
48
+ "cacheFingerprint": "5d56d8dd9ccc445000f843298d921611e777f90010f75bdfd344988a67b2f4bb",
49
+ "sourceManifestFingerprint": "91aa4345a785276c8ddd3afeb659f9418117121e4b03d12accdcb466351654e7",
50
+ "sourceManifestSha256": "98de0a2954e9111d8c1015a3b59afecd2b085a53078d53b13427c1a2e0e79544",
51
+ "sampling": {
52
+ "seed": 20260815,
53
+ "perClass": {
54
+ "train": 3000,
55
+ "valid": 500,
56
+ "test": 500
57
+ }
58
+ },
59
+ "splits": {
60
+ "train": {
61
+ "count": 120000,
62
+ "images": {
63
+ "filename": "train-images.npy",
64
+ "bytes": 491520128,
65
+ "sha256": "f0b13c42b667dc8b368ebbc24f7d154fdc7b2736128887c2b73cd3d848e4af2c",
66
+ "shape": [
67
+ 120000,
68
+ 64,
69
+ 64
70
+ ],
71
+ "dtype": "uint8"
72
+ },
73
+ "labels": {
74
+ "filename": "train-labels.npy",
75
+ "bytes": 240128,
76
+ "sha256": "cad1bef39f1b1578c1908ece8273676b650747ae12f6ba2ef7fff07e0518df76",
77
+ "shape": [
78
+ 120000
79
+ ],
80
+ "dtype": "int16"
81
+ }
82
+ },
83
+ "valid": {
84
+ "count": 20000,
85
+ "images": {
86
+ "filename": "valid-images.npy",
87
+ "bytes": 81920128,
88
+ "sha256": "e22c85d7741e7d661620c9db2b2acea9cbc01c025cd4371c45c2c12813e679e9",
89
+ "shape": [
90
+ 20000,
91
+ 64,
92
+ 64
93
+ ],
94
+ "dtype": "uint8"
95
+ },
96
+ "labels": {
97
+ "filename": "valid-labels.npy",
98
+ "bytes": 40128,
99
+ "sha256": "1c8d3d1400e77e2acf7caea884285c7b9b192c63b9ef24e5c7169f12c0ed298c",
100
+ "shape": [
101
+ 20000
102
+ ],
103
+ "dtype": "int16"
104
+ }
105
+ },
106
+ "test": {
107
+ "count": 20000,
108
+ "images": {
109
+ "filename": "test-images.npy",
110
+ "bytes": 81920128,
111
+ "sha256": "b08466e72d7494e664ac8f8b2f8b42857ac2e23ded846aa62e19430babad3a0d",
112
+ "shape": [
113
+ 20000,
114
+ 64,
115
+ 64
116
+ ],
117
+ "dtype": "uint8"
118
+ },
119
+ "labels": {
120
+ "filename": "test-labels.npy",
121
+ "bytes": 40128,
122
+ "sha256": "1c8d3d1400e77e2acf7caea884285c7b9b192c63b9ef24e5c7169f12c0ed298c",
123
+ "shape": [
124
+ 20000
125
+ ],
126
+ "dtype": "int16"
127
+ }
128
+ }
129
+ },
130
+ "rasterizerFiles": {
131
+ "worker": {
132
+ "path": "scripts/rasterize-stroke3-worker.mjs",
133
+ "sha256": "54bffd0ffc04c8d2958327a2206104676623d175f868c4e407774fd08db3431f"
134
+ },
135
+ "implementation": {
136
+ "path": "src/sketch-rasterizer.mjs",
137
+ "sha256": "73f93553d1961265f8c0a63519b91da736f2bd4f081d6699ed60ba3b74eafa59"
138
+ }
139
+ },
140
+ "builder": {
141
+ "path": "scripts/build-training-cache.py",
142
+ "sha256": "69bf16f53bf66bf850acd922e10c1ecafce68ee0fffb55cbce799da5aa181c93",
143
+ "dependencies": {
144
+ "model/integrity.py": "8b2aaa5ead6ec5739acaf279926deb553bc945eb8a3d5ccadb3fd20d3021c2a9"
145
+ },
146
+ "runtime": {
147
+ "python": "3.9.6",
148
+ "numpy": "1.26.4"
149
+ }
150
+ },
151
+ "node": {
152
+ "version": "v24.14.1",
153
+ "sha256": "91d3afb67a7dc211f642f2328f822505ff1c1ad381e98ffdd21b003970deeeda"
154
+ }
155
+ },
156
+ "rasterizer": {
157
+ "worker": "scripts/rasterize-stroke3-worker.mjs",
158
+ "size": 64,
159
+ "padding": 4,
160
+ "strokeWidth": 2.5,
161
+ "supersample": 4
162
+ },
163
+ "architecture": {
164
+ "inputShape": [
165
+ 1,
166
+ 64,
167
+ 64
168
+ ],
169
+ "numClasses": 40,
170
+ "channels": [
171
+ 32,
172
+ 64,
173
+ 128
174
+ ],
175
+ "embeddingSize": 192,
176
+ "dropout": 0.2
177
+ },
178
+ "model": {
179
+ "filename": "quickdraw-mvp.onnx",
180
+ "format": "ONNX",
181
+ "opset": 17,
182
+ "bytes": 1980135,
183
+ "sha256": "ba83ba933fa4d93399036f80ec3848257154f1e1661e0f1270beac17d0947c46",
184
+ "parameterCount": 494760,
185
+ "input": {
186
+ "name": "bitmap",
187
+ "shape": [
188
+ "batch",
189
+ 1,
190
+ 64,
191
+ 64
192
+ ],
193
+ "dtype": "float32",
194
+ "normalization": "uint8 / 255"
195
+ },
196
+ "output": {
197
+ "name": "logits",
198
+ "shape": [
199
+ "batch",
200
+ 40
201
+ ],
202
+ "dtype": "float32"
203
+ }
204
+ },
205
+ "training": {
206
+ "seed": 20260815,
207
+ "epochs": 8,
208
+ "bestEpoch": 8,
209
+ "batchSize": 256,
210
+ "optimizer": {
211
+ "name": "AdamW",
212
+ "learningRate": 0.001,
213
+ "weightDecay": 0.0001
214
+ },
215
+ "numWorkers": 0,
216
+ "minimumTestAccuracy": 0.7,
217
+ "minimumPerClassAccuracy": 0.7,
218
+ "deterministicAlgorithms": true,
219
+ "device": "mps",
220
+ "history": [
221
+ {
222
+ "epoch": 1,
223
+ "train": {
224
+ "loss": 1.056165291372935,
225
+ "accuracy": 0.7128833333333333,
226
+ "top5Accuracy": null,
227
+ "perClassAccuracy": null
228
+ },
229
+ "valid": {
230
+ "loss": 0.6744539570093155,
231
+ "accuracy": 0.8125,
232
+ "top5Accuracy": 0.9579,
233
+ "perClassAccuracy": [
234
+ 0.764,
235
+ 0.946,
236
+ 0.696,
237
+ 0.88,
238
+ 0.69,
239
+ 0.74,
240
+ 0.588,
241
+ 0.848,
242
+ 0.908,
243
+ 0.872,
244
+ 0.638,
245
+ 0.892,
246
+ 0.888,
247
+ 0.924,
248
+ 0.766,
249
+ 0.828,
250
+ 0.572,
251
+ 0.946,
252
+ 0.858,
253
+ 0.862,
254
+ 0.842,
255
+ 0.62,
256
+ 0.924,
257
+ 0.924,
258
+ 0.97,
259
+ 0.932,
260
+ 0.572,
261
+ 0.836,
262
+ 0.71,
263
+ 0.952,
264
+ 0.85,
265
+ 0.594,
266
+ 0.932,
267
+ 0.94,
268
+ 0.9,
269
+ 0.948,
270
+ 0.43,
271
+ 0.85,
272
+ 0.778,
273
+ 0.89
274
+ ]
275
+ }
276
+ },
277
+ {
278
+ "epoch": 2,
279
+ "train": {
280
+ "loss": 0.5841791171073913,
281
+ "accuracy": 0.835875,
282
+ "top5Accuracy": null,
283
+ "perClassAccuracy": null
284
+ },
285
+ "valid": {
286
+ "loss": 0.45889934716224673,
287
+ "accuracy": 0.8725,
288
+ "top5Accuracy": 0.97585,
289
+ "perClassAccuracy": [
290
+ 0.842,
291
+ 0.97,
292
+ 0.878,
293
+ 0.922,
294
+ 0.618,
295
+ 0.882,
296
+ 0.918,
297
+ 0.918,
298
+ 0.962,
299
+ 0.726,
300
+ 0.8,
301
+ 0.92,
302
+ 0.95,
303
+ 0.906,
304
+ 0.81,
305
+ 0.912,
306
+ 0.718,
307
+ 0.95,
308
+ 0.88,
309
+ 0.874,
310
+ 0.884,
311
+ 0.85,
312
+ 0.942,
313
+ 0.936,
314
+ 0.948,
315
+ 0.914,
316
+ 0.88,
317
+ 0.8,
318
+ 0.662,
319
+ 0.96,
320
+ 0.928,
321
+ 0.88,
322
+ 0.948,
323
+ 0.962,
324
+ 0.916,
325
+ 0.904,
326
+ 0.654,
327
+ 0.91,
328
+ 0.766,
329
+ 0.9
330
+ ]
331
+ }
332
+ },
333
+ {
334
+ "epoch": 3,
335
+ "train": {
336
+ "loss": 0.48913656921386717,
337
+ "accuracy": 0.8603083333333333,
338
+ "top5Accuracy": null,
339
+ "perClassAccuracy": null
340
+ },
341
+ "valid": {
342
+ "loss": 0.4167406298160553,
343
+ "accuracy": 0.8825,
344
+ "top5Accuracy": 0.9774,
345
+ "perClassAccuracy": [
346
+ 0.878,
347
+ 0.942,
348
+ 0.846,
349
+ 0.9,
350
+ 0.632,
351
+ 0.944,
352
+ 0.912,
353
+ 0.926,
354
+ 0.958,
355
+ 0.79,
356
+ 0.812,
357
+ 0.928,
358
+ 0.904,
359
+ 0.892,
360
+ 0.83,
361
+ 0.896,
362
+ 0.74,
363
+ 0.942,
364
+ 0.906,
365
+ 0.914,
366
+ 0.908,
367
+ 0.91,
368
+ 0.924,
369
+ 0.95,
370
+ 0.954,
371
+ 0.892,
372
+ 0.814,
373
+ 0.878,
374
+ 0.742,
375
+ 0.98,
376
+ 0.954,
377
+ 0.918,
378
+ 0.91,
379
+ 0.96,
380
+ 0.938,
381
+ 0.884,
382
+ 0.724,
383
+ 0.858,
384
+ 0.796,
385
+ 0.914
386
+ ]
387
+ }
388
+ },
389
+ {
390
+ "epoch": 4,
391
+ "train": {
392
+ "loss": 0.43742944836616515,
393
+ "accuracy": 0.8745666666666667,
394
+ "top5Accuracy": null,
395
+ "perClassAccuracy": null
396
+ },
397
+ "valid": {
398
+ "loss": 0.3852368968963623,
399
+ "accuracy": 0.88965,
400
+ "top5Accuracy": 0.9801,
401
+ "perClassAccuracy": [
402
+ 0.872,
403
+ 0.978,
404
+ 0.864,
405
+ 0.968,
406
+ 0.734,
407
+ 0.922,
408
+ 0.924,
409
+ 0.9,
410
+ 0.966,
411
+ 0.858,
412
+ 0.782,
413
+ 0.932,
414
+ 0.866,
415
+ 0.894,
416
+ 0.862,
417
+ 0.912,
418
+ 0.808,
419
+ 0.912,
420
+ 0.912,
421
+ 0.926,
422
+ 0.9,
423
+ 0.86,
424
+ 0.954,
425
+ 0.926,
426
+ 0.966,
427
+ 0.944,
428
+ 0.848,
429
+ 0.852,
430
+ 0.792,
431
+ 0.964,
432
+ 0.95,
433
+ 0.944,
434
+ 0.844,
435
+ 0.968,
436
+ 0.948,
437
+ 0.92,
438
+ 0.774,
439
+ 0.886,
440
+ 0.636,
441
+ 0.918
442
+ ]
443
+ }
444
+ },
445
+ {
446
+ "epoch": 5,
447
+ "train": {
448
+ "loss": 0.39647118004163107,
449
+ "accuracy": 0.885475,
450
+ "top5Accuracy": null,
451
+ "perClassAccuracy": null
452
+ },
453
+ "valid": {
454
+ "loss": 0.3675016505241394,
455
+ "accuracy": 0.8968,
456
+ "top5Accuracy": 0.97995,
457
+ "perClassAccuracy": [
458
+ 0.832,
459
+ 0.974,
460
+ 0.84,
461
+ 0.956,
462
+ 0.802,
463
+ 0.936,
464
+ 0.788,
465
+ 0.862,
466
+ 0.96,
467
+ 0.866,
468
+ 0.81,
469
+ 0.936,
470
+ 0.956,
471
+ 0.926,
472
+ 0.916,
473
+ 0.934,
474
+ 0.784,
475
+ 0.946,
476
+ 0.932,
477
+ 0.926,
478
+ 0.916,
479
+ 0.846,
480
+ 0.954,
481
+ 0.932,
482
+ 0.966,
483
+ 0.93,
484
+ 0.796,
485
+ 0.878,
486
+ 0.752,
487
+ 0.962,
488
+ 0.93,
489
+ 0.844,
490
+ 0.946,
491
+ 0.964,
492
+ 0.968,
493
+ 0.964,
494
+ 0.726,
495
+ 0.914,
496
+ 0.858,
497
+ 0.944
498
+ ]
499
+ }
500
+ },
501
+ {
502
+ "epoch": 6,
503
+ "train": {
504
+ "loss": 0.36662511706352235,
505
+ "accuracy": 0.892825,
506
+ "top5Accuracy": null,
507
+ "perClassAccuracy": null
508
+ },
509
+ "valid": {
510
+ "loss": 0.36927314329147337,
511
+ "accuracy": 0.8964,
512
+ "top5Accuracy": 0.98105,
513
+ "perClassAccuracy": [
514
+ 0.866,
515
+ 0.966,
516
+ 0.79,
517
+ 0.922,
518
+ 0.81,
519
+ 0.864,
520
+ 0.826,
521
+ 0.882,
522
+ 0.962,
523
+ 0.8,
524
+ 0.8,
525
+ 0.958,
526
+ 0.948,
527
+ 0.932,
528
+ 0.866,
529
+ 0.96,
530
+ 0.674,
531
+ 0.954,
532
+ 0.934,
533
+ 0.914,
534
+ 0.904,
535
+ 0.844,
536
+ 0.978,
537
+ 0.94,
538
+ 0.966,
539
+ 0.948,
540
+ 0.858,
541
+ 0.89,
542
+ 0.828,
543
+ 0.954,
544
+ 0.952,
545
+ 0.904,
546
+ 0.936,
547
+ 0.97,
548
+ 0.958,
549
+ 0.936,
550
+ 0.704,
551
+ 0.914,
552
+ 0.898,
553
+ 0.946
554
+ ]
555
+ }
556
+ },
557
+ {
558
+ "epoch": 7,
559
+ "train": {
560
+ "loss": 0.3465169951756795,
561
+ "accuracy": 0.897975,
562
+ "top5Accuracy": null,
563
+ "perClassAccuracy": null
564
+ },
565
+ "valid": {
566
+ "loss": 0.35224454931616783,
567
+ "accuracy": 0.9008,
568
+ "top5Accuracy": 0.9812,
569
+ "perClassAccuracy": [
570
+ 0.882,
571
+ 0.974,
572
+ 0.89,
573
+ 0.946,
574
+ 0.75,
575
+ 0.906,
576
+ 0.9,
577
+ 0.91,
578
+ 0.972,
579
+ 0.828,
580
+ 0.762,
581
+ 0.95,
582
+ 0.948,
583
+ 0.926,
584
+ 0.892,
585
+ 0.95,
586
+ 0.722,
587
+ 0.952,
588
+ 0.946,
589
+ 0.932,
590
+ 0.902,
591
+ 0.836,
592
+ 0.958,
593
+ 0.95,
594
+ 0.966,
595
+ 0.946,
596
+ 0.866,
597
+ 0.894,
598
+ 0.782,
599
+ 0.968,
600
+ 0.944,
601
+ 0.852,
602
+ 0.928,
603
+ 0.968,
604
+ 0.954,
605
+ 0.94,
606
+ 0.802,
607
+ 0.892,
608
+ 0.796,
609
+ 0.95
610
+ ]
611
+ }
612
+ },
613
+ {
614
+ "epoch": 8,
615
+ "train": {
616
+ "loss": 0.321421586004893,
617
+ "accuracy": 0.9052166666666667,
618
+ "top5Accuracy": null,
619
+ "perClassAccuracy": null
620
+ },
621
+ "valid": {
622
+ "loss": 0.34855738773345946,
623
+ "accuracy": 0.90335,
624
+ "top5Accuracy": 0.9813,
625
+ "perClassAccuracy": [
626
+ 0.882,
627
+ 0.976,
628
+ 0.888,
629
+ 0.97,
630
+ 0.862,
631
+ 0.914,
632
+ 0.908,
633
+ 0.904,
634
+ 0.97,
635
+ 0.92,
636
+ 0.85,
637
+ 0.9,
638
+ 0.938,
639
+ 0.93,
640
+ 0.912,
641
+ 0.898,
642
+ 0.77,
643
+ 0.912,
644
+ 0.904,
645
+ 0.932,
646
+ 0.876,
647
+ 0.922,
648
+ 0.956,
649
+ 0.906,
650
+ 0.95,
651
+ 0.936,
652
+ 0.872,
653
+ 0.858,
654
+ 0.814,
655
+ 0.956,
656
+ 0.902,
657
+ 0.912,
658
+ 0.938,
659
+ 0.964,
660
+ 0.89,
661
+ 0.94,
662
+ 0.806,
663
+ 0.916,
664
+ 0.734,
665
+ 0.946
666
+ ]
667
+ }
668
+ }
669
+ ]
670
+ },
671
+ "metrics": {
672
+ "test": {
673
+ "loss": 0.3516912404060364,
674
+ "accuracy": 0.9005,
675
+ "top5Accuracy": 0.98195,
676
+ "perClassAccuracy": [
677
+ 0.9,
678
+ 0.968,
679
+ 0.844,
680
+ 0.96,
681
+ 0.836,
682
+ 0.916,
683
+ 0.912,
684
+ 0.914,
685
+ 0.944,
686
+ 0.916,
687
+ 0.86,
688
+ 0.872,
689
+ 0.938,
690
+ 0.916,
691
+ 0.898,
692
+ 0.916,
693
+ 0.784,
694
+ 0.924,
695
+ 0.888,
696
+ 0.922,
697
+ 0.872,
698
+ 0.91,
699
+ 0.96,
700
+ 0.912,
701
+ 0.942,
702
+ 0.936,
703
+ 0.878,
704
+ 0.884,
705
+ 0.784,
706
+ 0.956,
707
+ 0.9,
708
+ 0.918,
709
+ 0.946,
710
+ 0.978,
711
+ 0.838,
712
+ 0.958,
713
+ 0.796,
714
+ 0.922,
715
+ 0.738,
716
+ 0.964
717
+ ]
718
+ },
719
+ "onnxTest": {
720
+ "loss": 0.35169129371643065,
721
+ "accuracy": 0.9005,
722
+ "top5Accuracy": 0.98195,
723
+ "perClassAccuracy": [
724
+ 0.9,
725
+ 0.968,
726
+ 0.844,
727
+ 0.96,
728
+ 0.836,
729
+ 0.916,
730
+ 0.912,
731
+ 0.914,
732
+ 0.944,
733
+ 0.916,
734
+ 0.86,
735
+ 0.872,
736
+ 0.938,
737
+ 0.916,
738
+ 0.898,
739
+ 0.916,
740
+ 0.784,
741
+ 0.924,
742
+ 0.888,
743
+ 0.922,
744
+ 0.872,
745
+ 0.91,
746
+ 0.96,
747
+ 0.912,
748
+ 0.942,
749
+ 0.936,
750
+ 0.878,
751
+ 0.884,
752
+ 0.784,
753
+ 0.956,
754
+ 0.9,
755
+ 0.918,
756
+ 0.946,
757
+ 0.978,
758
+ 0.838,
759
+ 0.958,
760
+ 0.796,
761
+ 0.922,
762
+ 0.738,
763
+ 0.964
764
+ ]
765
+ },
766
+ "maxLogitDifference": 4.00543212890625e-05
767
+ },
768
+ "runtime": {
769
+ "python": "3.9.6",
770
+ "platform": "macOS-26.6-arm64-arm-64bit",
771
+ "machine": "arm64",
772
+ "numpy": "1.26.4",
773
+ "torch": "2.8.0",
774
+ "onnx": "1.19.1",
775
+ "onnxruntime": "1.19.2"
776
+ },
777
+ "browserRuntime": {
778
+ "package": "onnxruntime-web",
779
+ "version": "1.22.0",
780
+ "lockfileSha256": "e50f2a9ff0a010abd2b3a25c0a54b8d5465602c4c47bcd037e716e9f1bd5ea5d"
781
+ },
782
+ "source": {
783
+ "repositoryRevision": "d2786f128e3b1ec07cfee1b4ef14ab797f53c026",
784
+ "filesMatchRevision": true,
785
+ "files": {
786
+ "model/__init__.py": "925a4d0bae99898cc145c54eb034f79f5d662ba8f7ae7854f540134c171088b0",
787
+ "model/quickdraw_cnn.py": "cb1906705fdae3d6cb67e1b897558fb01a040bf3400051d5c03d2709e1f19c8f",
788
+ "model/integrity.py": "8b2aaa5ead6ec5739acaf279926deb553bc945eb8a3d5ccadb3fd20d3021c2a9",
789
+ "scripts/build-training-cache.py": "69bf16f53bf66bf850acd922e10c1ecafce68ee0fffb55cbce799da5aa181c93",
790
+ "scripts/rasterize-stroke3-worker.mjs": "54bffd0ffc04c8d2958327a2206104676623d175f868c4e407774fd08db3431f",
791
+ "scripts/train-quickdraw-model.py": "3d208847fab3f2fb870a8fbea678cb3d7c2f797007a183623fab306405283ee5",
792
+ "src/sketch-rasterizer.mjs": "73f93553d1961265f8c0a63519b91da736f2bd4f081d6699ed60ba3b74eafa59",
793
+ "package.json": "de13417c20a49a170d66856b17941731a64fad6913d1badeb4de1509dd396f82",
794
+ "package-lock.json": "e50f2a9ff0a010abd2b3a25c0a54b8d5465602c4c47bcd037e716e9f1bd5ea5d",
795
+ "requirements-data.txt": "e77e7045d01a9431264f9262c6b199d8e7988c139276a5266a3cf689588ad66e",
796
+ "requirements-model.txt": "58d3cbb9c136a64c66502affdf57da0b0cd2a911332059a5f51eda5a3e7770f6"
797
+ }
798
+ },
799
+ "fingerprint": "e0c9f0e3edabeb2d9f80c00e95ac311b906eecac5aced2a89730c2d10ff15602"
800
+ }
models/quickdraw-mvp/quickdraw-mvp.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ba83ba933fa4d93399036f80ec3848257154f1e1661e0f1270beac17d0947c46
3
+ size 1980135
models/sketch-embedder/model.json ADDED
@@ -0,0 +1,289 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schemaVersion": 1,
3
+ "kind": "sketch-embedder",
4
+ "classes": [
5
+ "airplane",
6
+ "apple",
7
+ "banana",
8
+ "bicycle",
9
+ "bird",
10
+ "book",
11
+ "cake",
12
+ "camera",
13
+ "car",
14
+ "cat",
15
+ "chair",
16
+ "clock",
17
+ "cloud",
18
+ "coffee cup",
19
+ "computer",
20
+ "door",
21
+ "eye",
22
+ "face",
23
+ "fish",
24
+ "flower",
25
+ "hamburger",
26
+ "hand",
27
+ "house",
28
+ "ice cream",
29
+ "key",
30
+ "light bulb",
31
+ "moon",
32
+ "mountain",
33
+ "pencil",
34
+ "pizza",
35
+ "rain",
36
+ "rainbow",
37
+ "star",
38
+ "sun",
39
+ "tree",
40
+ "truck",
41
+ "umbrella"
42
+ ],
43
+ "trainingData": {
44
+ "sketchCacheFingerprint": "5d56d8dd9ccc445000f843298d921611e777f90010f75bdfd344988a67b2f4bb",
45
+ "iconCacheFingerprint": "94a2d3e022c8d482df631fa87756ea7a19e768bb829caa853ca78c2430a62e8b",
46
+ "iconCounts": {
47
+ "planned": 480,
48
+ "kept": 479,
49
+ "train": 439,
50
+ "holdout": 40,
51
+ "byStatus": {
52
+ "ok": 479,
53
+ "unreadable": 1,
54
+ "empty": 0,
55
+ "inkOutOfBand": 0
56
+ },
57
+ "minPerClass": 11,
58
+ "sparseClasses": []
59
+ },
60
+ "source": {
61
+ "repository": "https://github.com/nsdevaraj/SVGDepot.git",
62
+ "commit": "5d05c6fa39b9d193d408d203e72f98e3e78c5b3b"
63
+ }
64
+ },
65
+ "model": {
66
+ "filename": "sketch-embedder.onnx",
67
+ "format": "ONNX",
68
+ "opset": 17,
69
+ "bytes": 2009324,
70
+ "sha256": "3243c8679464f88431cf061dfd632843ec3a5377ff17999e126f0425d9e3713a",
71
+ "parameterCount": 501798,
72
+ "embeddingDim": 64,
73
+ "input": {
74
+ "name": "bitmap",
75
+ "dtype": "float32",
76
+ "shape": [
77
+ "batch",
78
+ 1,
79
+ 64,
80
+ 64
81
+ ],
82
+ "normalization": "uint8 / 255"
83
+ },
84
+ "outputs": {
85
+ "embedding": {
86
+ "name": "embedding",
87
+ "dtype": "float32",
88
+ "shape": [
89
+ "batch",
90
+ 64
91
+ ],
92
+ "normalization": "L2"
93
+ },
94
+ "logits": {
95
+ "name": "logits",
96
+ "dtype": "float32",
97
+ "shape": [
98
+ "batch",
99
+ 37
100
+ ]
101
+ }
102
+ }
103
+ },
104
+ "rasterizer": {
105
+ "worker": "scripts/rasterize-svg-worker.mjs",
106
+ "size": 64,
107
+ "padding": 4,
108
+ "strokeWidth": 2.5,
109
+ "supersample": 4,
110
+ "minInk": 0.01,
111
+ "maxInk": 0.6
112
+ },
113
+ "metrics": {
114
+ "classification": {
115
+ "accuracy": 0.9022162162162162,
116
+ "top5Accuracy": 0.9772972972972973,
117
+ "perClassAccuracy": [
118
+ 0.904,
119
+ 0.948,
120
+ 0.754,
121
+ 0.868,
122
+ 0.788,
123
+ 0.826,
124
+ 0.936,
125
+ 0.962,
126
+ 0.846,
127
+ 0.832,
128
+ 0.914,
129
+ 0.958,
130
+ 0.924,
131
+ 0.882,
132
+ 0.878,
133
+ 0.956,
134
+ 0.902,
135
+ 0.856,
136
+ 0.912,
137
+ 0.892,
138
+ 0.952,
139
+ 0.924,
140
+ 0.972,
141
+ 0.948,
142
+ 0.91,
143
+ 0.88,
144
+ 0.79,
145
+ 0.972,
146
+ 0.87,
147
+ 0.896,
148
+ 0.938,
149
+ 0.956,
150
+ 0.946,
151
+ 0.934,
152
+ 0.9,
153
+ 0.894,
154
+ 0.962
155
+ ]
156
+ },
157
+ "retrieval": {
158
+ "all": {
159
+ "recallAt1": 0.8878378378378379,
160
+ "recallAt5": 0.918,
161
+ "recallAt10": 0.9308648648648649,
162
+ "perClassRecallAt10": [
163
+ 0.914,
164
+ 0.964,
165
+ 0.802,
166
+ 0.928,
167
+ 0.9,
168
+ 0.88,
169
+ 0.942,
170
+ 0.962,
171
+ 0.908,
172
+ 0.874,
173
+ 0.944,
174
+ 0.968,
175
+ 0.952,
176
+ 0.906,
177
+ 0.89,
178
+ 0.974,
179
+ 0.92,
180
+ 0.898,
181
+ 0.952,
182
+ 0.914,
183
+ 0.96,
184
+ 0.952,
185
+ 0.98,
186
+ 0.954,
187
+ 0.948,
188
+ 0.896,
189
+ 0.846,
190
+ 0.978,
191
+ 0.902,
192
+ 0.914,
193
+ 0.958,
194
+ 0.976,
195
+ 0.972,
196
+ 0.97,
197
+ 0.958,
198
+ 0.926,
199
+ 0.96
200
+ ]
201
+ },
202
+ "holdoutIcons": {
203
+ "recallAt1": 0.7225405405405405,
204
+ "recallAt5": 0.8932972972972973,
205
+ "recallAt10": 0.9657297297297297,
206
+ "perClassRecallAt10": [
207
+ 0.966,
208
+ 0.98,
209
+ 0.986,
210
+ 0.984,
211
+ 0.986,
212
+ 0.984,
213
+ 0.98,
214
+ 0.988,
215
+ 0.986,
216
+ 0.986,
217
+ 0.936,
218
+ 0.876,
219
+ 0.986,
220
+ 0.962,
221
+ 0.97,
222
+ 0.994,
223
+ 0.956,
224
+ 0.968,
225
+ 0.988,
226
+ 0.986,
227
+ 0.986,
228
+ 0.726,
229
+ 0.996,
230
+ 0.984,
231
+ 0.974,
232
+ 0.984,
233
+ 0.978,
234
+ 0.982,
235
+ 0.75,
236
+ 0.976,
237
+ 0.99,
238
+ 0.998,
239
+ 0.994,
240
+ 0.992,
241
+ 0.994,
242
+ 0.992,
243
+ 0.988
244
+ ]
245
+ }
246
+ },
247
+ "maxLogitDifference": 2.9727816581726074e-06
248
+ },
249
+ "training": {
250
+ "epochs": 8,
251
+ "batchSize": 256,
252
+ "optimizer": {
253
+ "name": "AdamW",
254
+ "learningRate": 0.001,
255
+ "weightDecay": 0.0001
256
+ },
257
+ "temperature": 0.07,
258
+ "classificationWeight": 1.0,
259
+ "seed": 20260815,
260
+ "deterministicAlgorithms": true,
261
+ "minClassAccuracy": 0.7,
262
+ "minRecallAt10": 0.7,
263
+ "minClassRecallAt10": 0.7,
264
+ "weakClasses": {
265
+ "byRecall": [],
266
+ "byClassification": []
267
+ }
268
+ },
269
+ "runtime": {
270
+ "python": "3.9.6",
271
+ "numpy": "1.26.4",
272
+ "torch": "2.8.0",
273
+ "onnx": "1.19.1",
274
+ "onnxruntime": "1.19.2"
275
+ },
276
+ "source": {
277
+ "files": {
278
+ "model/sketch_embedder.py": "910cba4fca9234231b52ee216c3a3c1d7d4bea544e8dea2de5816d822c9b6ae0",
279
+ "model/quickdraw_cnn.py": "cb1906705fdae3d6cb67e1b897558fb01a040bf3400051d5c03d2709e1f19c8f",
280
+ "model/integrity.py": "8b2aaa5ead6ec5739acaf279926deb553bc945eb8a3d5ccadb3fd20d3021c2a9",
281
+ "scripts/build-embedding-cache.py": "94413b5a1dc47de92e5fe978ad8fe9d91bc16453b2b7c2102f2c46250177f6c5",
282
+ "scripts/rasterize-svg-worker.mjs": "6f3ae6a04714b4dc35ad07deac234fc297f1d9e931ede6c34e17cfd1eeaeece9",
283
+ "scripts/train-sketch-embedder.py": "baddd6b1f60c8102e42bc59607841900e148e9e90798a7c5a703e7a159c6de67",
284
+ "src/svg-to-polylines.mjs": "b4f994fc6509167db99fdfecad7f400abdf98eb5cbb8c9857e0ec01020702865",
285
+ "src/sketch-rasterizer.mjs": "73f93553d1961265f8c0a63519b91da736f2bd4f081d6699ed60ba3b74eafa59"
286
+ }
287
+ },
288
+ "fingerprint": "fecbbbde3387ea29916ac4135167ccfb18b8a23e8a1d28a851df3b3c60845dc8"
289
+ }
models/sketch-embedder/sketch-embedder.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3243c8679464f88431cf061dfd632843ec3a5377ff17999e126f0425d9e3713a
3
+ size 2009324