gpu-doodle:一个跑在浏览器里的涂鸦识别小模型。
- 边画边猜,第一笔画个圆它会在月亮、时钟、太阳、苹果之间摇摆,第二笔下去就定了
- 100 个 Quick, Draw! 类别,测试集 top-1 89.4%
- 整个包(int6 权重 + 手写 WGSL 内核)Brotli 后 26 KB,3.2 万参数
- 有 WebGPU 就走 GPU,没有就走 CPU,任何现代浏览器都能玩
- 自带 20 秒限时的"你画我猜"模式,中英文界面跟随系统
训练数据是 Google 的 Quick, Draw!(CC BY 4.0),做法参考了 arikchakma 的 gpu-time。
Demo:
仓库:
显示更多