tfjs 移植到微信小程序

TensorFlow.js

TensorFlow.js 是谷歌出品的开源 JavaScript 库,能够用于训练和部署机器学习模型。TensorFlow.js 可以为你提供高性能的、易于使用的机器学习构建模块,允许你在浏览器上训练模型,或以推断模式运行预训练的模型。TensorFlow.js 不仅可以提供低级的机器学习构建模块,还可以提供高级的类似 Keras 的 API 来构建神经网络。

心血来潮

前段时间,我司一位持续创业的大佬突然心血来潮,看到 tfjs 官网出了几款 AI 小游戏(比如说 这个),很感兴趣想移植到小程序端火一把(或者说创业?)。

我对这块还是比较感兴趣的,所以主动接了这活,尝试把 tfjs 移植到小程序上。过程当然是极其不顺利的,因为小程序屏蔽了很多原生的 WebAPI,其中不少是被 tfjs 所依赖的。

柳暗花明又一村

移植的过程堪比魔改过程,也可以理解为 Debug 的过程,哪里报错改哪里,一步步修复兼容性问题。好在最后改下来,也就 tfjs-core 改动比较多,tfjs-converter 基本没动。

目的是要实现 示例 的功能,主要是实现了被小程序屏蔽的 fetch 方法,还有一些代码上的小调整。具体的可以看下 代码

移植成功后,把生成的 wx-tfjs-core 给了大佬,但是他似乎比较忙,也就没有「火一把」的下文了。欢迎有兴趣的同学继续接力,开发出有意思的 AI 小程序。如果有问题,欢迎交流。