前端机器学习TensorFlow.js里的预训练模型怎么使用?
一、问题描述
最近做项目有一个这样的需求,大概意思就是实现一个调用摄像头进行拍照,生成卡通等形象。然后其中有一个需求是在视频画面实时给人添加一些模型,比如人脸检测、身体分割、姿势估计等。那么就想到了用TensorFlow.js去实现这个需求。
经过一番了解,发现这个东东简直就是前端人员的巨大福利啊,在浏览器直接搞机器学习,太爽了。然后这篇文章我想介绍的是怎样去使用官方提供的一些预训练模型。
自己研究了一天,最终还是在我师兄的帮助下,才搞出来,上网查了很久,没有一篇去讲解这个东西,所以特此记录,希望下一个IT人少花点时间,去快速的学会怎样用官方提供的一些预训练模型。
二、TensorFlow.js官网
TensorFlow.js官网
进去是这样子的
你可以在模型中按照自己的需求选择。
当然这些操作很容易,我想说的是怎样用,怎样将模型里的demo在你电脑本地跑起来。
三、本地使用
1、到官网下载需要的模型
点击上图对应的模型就会进入该项目的开源github地址,然后把代码弄下来,具体怎么弄我就不说了。很简单,不会去百度就可以。
2、下载好之后解压
注意:解压目录文件名不要起中文名字,很容易出问题。
解压后
3、在本地运行你需要的模型
这篇文章以姿势估计(PoseNet)为例,其余模型基本操作和这个都一样。