TensorFlow.js是一个 JavaScript 库,用于在浏览器和 Node.js 上训练和部署机器学习模型。
同样的,现在也可以在微信小程序里使用 TensorFlow 团队提供的插件,运行一些开箱即用的预训练模型。
现在项目中已经引入:实时估计人体姿势(PoseNet)和定位和识别单个图像中的多个对象(Coco SSD)两个模型。
-
首先需要在小程序后台添加
TensorFlow.js插件,参考此文档。 -
在项目根目录下安装项目需要用到的
npm包,使用yarn或者npm都可。
yarn install-
注意:安装好
npm包后一定有在开发者工具里构建npm。可以参考微信官方文档,在小程序中使用npm。 -
修改根目录下的
env.js.example文件,把其中的模型地址替换成你的模型地址。
微信搜索:TensorFlow机器学习模型。或者扫码:
