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

本文详细介绍了如何在本地使用TensorFlow.js的预训练模型,如PoseNet。通过从官网下载模型,解压并遵循官方教程在本地运行,包括设置环境、运行命令和解决可能出现的问题,例如依赖安装、文件修改等。按照本文步骤,10分钟即可完成预训练模型的本地运行。
摘要由CSDN通过智能技术生成

一、问题描述

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

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

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

二、TensorFlow.js官网

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

三、本地使用

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

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

2、下载好之后解压

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

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

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

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

  • 3
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值