当前位置:   article > 正文

前端机器学习TensorFlow.js里的预训练模型怎么使用?_前端模型训练

前端模型训练

一、问题描述

最近做项目有一个这样的需求,大概意思就是实现一个调用摄像头进行拍照,生成卡通等形象。然后其中有一个需求是在视频画面实时给人添加一些模型,比如人脸检测、身体分割、姿势估计等。那么就想到了用TensorFlow.js去实现这个需求。

经过一番了解,发现这个东东简直就是前端人员的巨大福利啊,在浏览器直接搞机器学习,太爽了。然后这篇文章我想介绍的是怎样去使用官方提供的一些预训练模型。

自己研究了一天,最终还是在我师兄的帮助下,才搞出来,上网查了很久,没有一篇去讲解这个东西,所以特此记录,希望下一个IT人少花点时间,去快速的学会怎样用官方提供的一些预训练模型。

二、TensorFlow.js官网

TensorFlow.js官网
进去是这样子的
在这里插入图片描述
你可以在模型中按照自己的需求选择。
当然这些操作很容易,我想说的是怎样用,怎样将模型里的demo在你电脑本地跑起来。

三、本地使用

1、到官网下载需要的模型

点击上图对应的模型就会进入该项目的开源github地址,然后把代码弄下来,具体怎么弄我就不说了。很简单,不会去百度就可以。
在这里插入图片描述

2、下载好之后解压

注意:解压目录文件名不要起中文名字,很容易出问题。
在这里插入图片描述
解压后
在这里插入图片描述

3、在本地运行你需要的模型

这篇文章以姿势估计(PoseNet)为例,其余模型基本操作和这个都一样。

1、点开github中对应的项目的demo,浏览使用步骤

在这里插入图片描述

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/爱喝兽奶帝天荒/article/detail/802779
推荐阅读
相关标签
  

闽ICP备14008679号