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

2020-10-24 03:47:53 +08:00
 lanking520

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

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

可以来看看这个项目

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

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

9067 次点击
所在节点    Android
14 条回复
user8341
2020-10-24 06:20:46 +08:00
很不错啊。魔改了什么地方呢？
lanking520
2020-10-24 07:29:52 +08:00
@user8341 主要是 Model 本身，把 channel 从 3 换成 1 了，减少计算量
rming
2020-10-24 07:34:47 +08:00
有点意思
airqj
2020-10-24 09:34:30 +08:00
不如做成小程序
user8341
2020-10-24 10:08:08 +08:00
@airqj 小程序能跑 AI ？
nnqijiu
2020-10-24 10:11:20 +08:00
楼主的涂鸦训练数据集是怎么获取的？
airqj
2020-10-24 10:45:54 +08:00
@user8341 tensorflow.js 支持微信小程序一年多了
Merlini
2020-10-24 10:50:26 +08:00
很有意思，希望以后我也能写个自然语言处理的 app
user8341
2020-10-24 11:26:24 +08:00
@airqj 这么厉害。我孤陋寡闻了。
no13bus
2020-10-24 12:14:42 +08:00
很赞
Escapist367
2020-10-24 13:10:16 +08:00
为什么要本地运算，而不是把模型部署在服务器上？
lanking520
2020-10-24 15:11:50 +08:00
@Escapist367 小模型直接跑，安卓就够用了，服务器还得花钱，还有一个延迟问题
lanking520
2020-10-24 15:12:07 +08:00
@nnqijiu QuickDraw 数据集，也可以用 kaggle 的
lanking520
2020-10-24 15:12:32 +08:00
@user8341 小程序的问题是没法用设备本身的算力

这是一个专为移动设备优化的页面（即为了让你能够在 Google 搜索结果里秒开这个页面），如果你希望参与 V2EX 社区的讨论，你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/718073

V2EX 是创意工作者们的社区，是一个分享自己正在做的有趣事物、交流想法，可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX