上个月完成了一个你画我猜的 APP,和几年前的 Google 出的《猜画小歌》很像。只是这次全部的推理部分都搬到 Android 上面。
模型用了魔改的 PyTorch MobileNetV2,轻量级模型,精确度达到 top1 79%, top3 91%。然后能识别 300 多种手写涂鸦。用的数据集是 QuickDraw 。推理部署十分简单,推理的速度也是非常的快。用红米 Note10 测试基本无卡顿。
可以来看看这个项目
项目地址 https://github.com/aws-samples/djl-demo/tree/master/android
用的是 DJL 基于 Java 的深度学习框架
1
user8341 2020-10-24 06:20:46 +08:00
很不错啊。魔改了什么地方呢?
|
2
lanking520 OP @user8341 主要是 Model 本身,把 channel 从 3 换成 1 了,减少计算量
|
3
rming 2020-10-24 07:34:47 +08:00
有点意思
|
4
airqj 2020-10-24 09:34:30 +08:00 via Android
不如做成小程序
|
6
nnqijiu 2020-10-24 10:11:20 +08:00
楼主的涂鸦训练数据集是怎么获取的?
|
8
Merlini 2020-10-24 10:50:26 +08:00
很有意思,希望以后我也能写个自然语言处理的 app
|
10
no13bus 2020-10-24 12:14:42 +08:00
很赞
|
11
Escapist367 2020-10-24 13:10:16 +08:00 via Android
为什么要本地运算,而不是把模型部署在服务器上?
|
12
lanking520 OP @Escapist367 小模型直接跑,安卓就够用了,服务器还得花钱,还有一个延迟问题
|
13
lanking520 OP @nnqijiu QuickDraw 数据集,也可以用 kaggle 的
|
14
lanking520 OP @user8341 小程序的问题是没法用设备本身的算力
|