推荐一个《涂鸦识别》的 AI 安卓项目

日期 : 2020-10-24 22:19:46作者 : 冰淇淋

上个月完成了一个你画我猜的 APP,和几年前的 Google 出的《猜画小歌》很像。只是这次全部的推理部分都搬到 Android 上面。

模型用了魔改的 PyTorch MobileNetV2,轻量级模型,精确度达到 top1 79%, top3 91%。然后能识别 300 多种手写涂鸦。用的数据集是 QuickDraw 。推理部署十分简单,推理的速度也是非常的快。用红米 Note10 测试基本无卡顿。

可以来看看这个项目

项目地址 https://github.com/aws-samples/djl-demo/tree/master/android

用的是 DJL 基于 Java 的深度学习框架

标签 :