推荐一个《涂鸦识别》的 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 的深度学习框架

4911 次点击
所在节点    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