你比个手势,AI就识别出Emoji,浏览器上跑:已开源

你比个手势,AI就识别出Emoji,浏览器上跑:已开源

本文经AI新媒体量子位(公众号ID:QbitAI)授权转载,转载请联系出处。

一只手势识别AI,怎样才算得上 (优) 秀啊?

不停地变换姿势,都能实时输出Emoji,大约是很秀了:

是守卫 (误) ,是来自《星际旅行》瓦肯的举手礼。不大常用,做起来甚至有难度。

是好运,一般会两只手一起比。不过,也不是人类通用的手势。

即便这样,AI依然机智地识别出来。并且,它是在浏览器上跑,也几乎没有延时

AI的爸爸,名字叫Nick Bourdakos (简称“尼克”) ,是来自IBM的程序猿。

尼克把自己的调教成果发了推特,揽下2.8万赞

尼克用的是TensorFlow.js,实时识别毫无压力。

他把算法开源了,说大家都可以试一试。

半小时就好

尼克说这个模型很简单,就是SSD-MobileNet

MobileNet是分类,SSD是目标检测,搭配食用也是常规方法。

他是用IBM云上的GPU训练的,免费的k80,半小时就训练好了。

在训练开始之前,要先准备数据:AI吃的是标注过的手势图

准备就绪,就来安装模型吧:

1$ npm install -g cloud-annotations 

然后,可以开始训练了:

 1$ cacli 


 2┌─────────────────────────────┐ 


 3│ (C)loud (A)nnotations (CLI) │ 


 4│ version 1.0.12 │ 


 5└─────────────────────────────┘ 


 6 


 7Usage: cacli <command> 


 8 


 9where <command> is one of: 


10 init Interactively create a config.yaml file 


11 train Start a training run 


12 logs Monitor the logs of a training run 


13 progress Monitor the progress of a training run 


14 list List all training runs 


15 download Download a trained model 


16 


17cacli <cmd> -h quick help on <cmd> 

当然,也不是非要用IBM云,也不是非要用GPU。拿CPU也能调教AI,大概要几小时吧。

训练完成,该在浏览器上跑了。GitHub项目里,自带了转换为TensorFlow.js模型的脚本。

把模型添加到React App里面。

最后,写一句nmp start,用浏览器打开http://localhost:3000。

耶,这样就可以对着屏幕随意舞动手指了,你的AI会明白的:

当然,这只机智的AI,才不是只能识别手指。

喝点什么

只看你用什么样的数据去投喂AI了。

曾经,尼克就帮AI修炼了分辨汽水的眼力。

第一题:一瓶雪碧,一瓶Canada Dry,都是绿色。

不管调换位置、还是侧过瓶身,AI都不会被迷惑。定格一看:

第二题:加大难度,两瓶都是Mountain Dew,一瓶普通一瓶低糖。

AI依然分得清楚,毫不犹豫。

分辨手势,分辨汽水,都不失水准。

那么问题来了,你想让AI识别什么呢?

想好了就开始调教吧,代码在这里:

相关推荐