【已开源】在小程序中运行TensorFlow机器学习模型
发布于 4 年前 作者 tqiao 5150 次浏览 来自 分享

背景

TensorFlow.js是一个 JavaScript 库,用于在浏览器和 Node.js 上训练和部署机器学习模型。

同样的,现在也可以在微信小程序里使用 TensorFlow 团队提供的插件,运行一些开箱即用的预训练模型。

现在项目中已经引入:实时估计人体姿势(PoseNet)和定位和识别单个图像中的多个对象(Coco SSD)两个模型。

快速开始

首先需要在小程序后台添加`` TensorFlow.js ``插件,<a href="https://mp.weixin.qq.com/wxopen/plugindevdoc?appid=wx6afed118d9e81df9&amp;token=378013697&amp;lang=zh_CN" rel="noopener" target="_blank">参考此文档。</a>
在项目根目录下安装项目需要用到的`` npm ``包,使用`` yarn ``或者`` npm ``都可。
yarn install
注意:安装好`` npm ``包后一定有在开发者工具里__构建`` npm ``__。可以参考微信官方文档,<a href="https://developers.weixin.qq.com/miniprogram/dev/devtools/npm.html" rel="noopener" target="_blank">在小程序中使用npm</a>。
修改根目录下的`` env.js.example ``文件,把其中的模型地址替换成你的模型地址。

线上版本

微信搜索:TensorFlow机器学习模型。或者扫码:

GitHub地址:https://github.com/GeekYmm/tensorflow-wxapp

1 回复

能否把检测过程也录制下来。谢谢了大神。

回到顶部