用 Tensorflow.js 来判断你女朋友的情绪

2020-10-16 11:50:43 浏览数 (1)

恩,这是一个很简单的表情识别——实际上就是入门级别的图片分类。使用方法就是在不同类别录入你女朋友的表情,然后在训练完毕后,会及时判断现在你女朋友的表情,再也不用猜她的想法了。

灵感来自 google 的一个网页 demo,不过这个 demo 是用 deaplearn.js ( tfjs 的前身)写的

楼主平时用 python 写 tensorflow 模型,看到今年 TensorFlow Dev Summit 推出的 tensorflow.js 后就想试试。网页用的是 AngularJS 写的,前端很久没碰过了,所以写得挺偷懒的。话说有 Typescript 加持后,写一些简单的网络还是蛮轻松的,不过也遇上一些坑,比如 js 版的 tensorflow 所有张量在使用过后会立即销毁,之前不知道这个特性,被坑了好久。

模型用的是卷积神经网络,使用摄像头采集对应情绪的照片来训练模型。实际效果不是特别好,因为我是把摄像头图片压缩到 32 * 32 的小图片送入网络的,所以实际上你能想象表情信息没有多少,但是你可以尝试一些比如转头,举手,这类特征明显的动作,还有是在录制的几秒钟时间里的动作大一点,这样分类的效果会比较好(本身每类样本就只有 10 个,太容易过拟合了)。

体验地址在这里: https://moevis.github.io/what-is-my-girlfriend-thinking/

什么?你没女朋友?哈哈哈哈哈

0 人点赞