讲师中心 微信公众号
AI工具推荐 视频效率加速

如何用TensorFlow.js在浏览器中运行机器学习模型?

冬杰小哥_5852

冬杰小哥_5852

发布时间:2025-09-29 19:24:02

|

741人浏览过

|

来源于php中文网

原创

通过TensorFlow.js可在浏览器中直接运行机器学习模型,1. 引入CDN脚本加载库;2. 加载预训练模型如MobileNet;3. 将图像转为张量并推理;4. 使用dispose或tidy管理内存,实现高效、隐私安全的前端AI推理。

如何用tensorflow.js在浏览器中运行机器学习模型?

在浏览器中运行机器学习模型可以通过 TensorFlow.js 实现,它允许你直接在前端加载和推理预训练模型,无需后端支持。整个过程不依赖服务器计算,响应快且保护用户隐私。

1. 引入 TensorFlow.js

最简单的方式是通过 CDN 在 HTML 文件中引入 TensorFlow.js:

<script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs"></script>

引入后,你就可以在 JavaScript 中使用 tf 全局变量进行模型操作和张量计算。

2. 加载预训练模型

你可以加载已经转换为 TensorFlow.js 格式的模型。常见来源包括:

  • TensorFlow.js 官方提供的模型(如 coco-ssd、mobilenet)
  • 自己用 Python 训练后导出的模型,再用 tfjs-converter 转换

以加载 MobileNet 为例:

import * as tf from '@tensorflow/tfjs';
const model = await tf.loadLayersModel('model.json');

或使用官方封装好的模型:

Expanso yaml-to-json
Expanso yaml-to-json

使用 Expanso Edge 管道将 YAML 输入转换为 JSON 格式。

下载
import * as mobilenet from '@tensorflow/tfjs-models/mobilenet';
const net = await mobilenet.load();

3. 预处理输入并执行推理

大多数图像模型需要将图像转换为张量。例如,从一个 img 元素获取数据:

const img = document.getElementById('myImage');
const tensor = tf.browser.fromPixels(img)
  .resizeNearestNeighbor([224, 224])
  .toFloat()
  .expandDims();

然后执行预测:

const prediction = await net.classify(tensor);
console.log(prediction);

对于自定义模型,使用 model.predict(tensor) 获取输出。

4. 释放资源

张量和中间结果会占用显存,推理完成后应手动清理:

tensor.dispose();
// 或批量清理
tf.dispose([tensor, output]);

也可以使用 tf.tidy() 自动管理内存:

tf.tidy(() => {
  const result = model.predict(input);
  return result;
});

基本上就这些。只要模型已转成 Web 友好格式,TensorFlow.js 能在多数现代浏览器中高效运行推理任务,适合做图像分类、姿态检测、文本生成等轻量级 AI 功能。关键点是模型格式正确、输入预处理匹配训练方式,以及注意内存管理。

热门AI工具

更多
音述AI
音述AI Hot

一款AI音频处理工具,主要用于音述AI是一个以“用声音述说故事”为核心的 AI 音乐创作与声音分享社区,适合需要提升相关任务效率的用户。

SkildArt
SkildArt Hot

SkildArt是一款AI文本写作工具,一站式 AI 视觉创作平台。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

咔片AIPPT

一款在线AI演示文稿制作工具,可根据主题和内容需求辅助生成PPT结构与页面,提高演示材料制作效率。

Loomy
Loomy Hot

一款AI工具,主要用于科大讯飞发布的桌面级 AI 助理,比 OpenClaw 更易用、更安全!,适合需要提升相关任务效率的用户。

PixTV
PixTV Hot

PixTV是一款面向AIGC内容创作的AI视频生成工具。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

VibeKnow
VibeKnow Hot

一款AI视频创作工具,主要用于全球首个AI知识视频创作平台,文档、文章、网页,一键生成视频,适合需要提升相关任务效率的用户。

相关专题

更多
json数据格式
json数据格式

JSON是一种轻量级的数据交换格式。本专题为大家带来json数据格式相关文章,帮助大家解决问题。

2275

2023.08.07

json是什么
json是什么

JSON是一种轻量级的数据交换格式,具有简洁、易读、跨平台和语言的特点,JSON数据是通过键值对的方式进行组织,其中键是字符串,值可以是字符串、数值、布尔值、数组、对象或者null,在Web开发、数据交换和配置文件等方面得到广泛应用。本专题为大家提供json相关的文章、下载、课程内容,供大家免费下载体验。

3722

2023.08.23

jquery怎么操作json
jquery怎么操作json

操作的方法有:1、“$.parseJSON(jsonString)”2、“$.getJSON(url, data, success)”;3、“$.each(obj, callback)”;4、“$.ajax()”。更多jquery怎么操作json的详细内容,可以访问本专题下面的文章。

1236

2023.10.13

go语言处理json数据方法
go语言处理json数据方法

本专题整合了go语言中处理json数据方法,阅读专题下面的文章了解更多详细内容。

4299

2025.09.10

c语言const用法
c语言const用法

const是关键字,可以用于声明常量、函数参数中的const修饰符、const修饰函数返回值、const修饰指针。详细介绍:1、声明常量,const关键字可用于声明常量,常量的值在程序运行期间不可修改,常量可以是基本数据类型,如整数、浮点数、字符等,也可是自定义的数据类型;2、函数参数中的const修饰符,const关键字可用于函数的参数中,表示该参数在函数内部不可修改等等。

2398

2023.09.20

全局变量怎么定义
全局变量怎么定义

本专题整合了全局变量相关内容,阅读专题下面的文章了解更多详细内容。

5145

2025.09.18

python 全局变量
python 全局变量

本专题整合了python中全局变量定义相关教程,阅读专题下面的文章了解更多详细内容。

2190

2025.09.18

c++ 全局变量
c++ 全局变量

本专题整合了c++全局变量的使用、定义、作用域等等内容,阅读专题下面的文章了解更多详细内容。

199

2026.03.17

Postman身份认证怎么配置
Postman身份认证怎么配置

本专题从实际API接口调试场景出发,讲解Postman身份认证的完整配置流程,包括常见认证协议介绍、授权参数设置、访问令牌填写、OAuth 2.0授权及认证失败排查,帮助用户正确完成接口授权与访问验证。

0

2026.11.06

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
WebStorm 官方调试文档
WebStorm 官方调试文档

共0课时 | 0人学习

React 教程
React 教程

共58课时 | 12.5万人学习

TypeScript 教程
TypeScript 教程

共19课时 | 6.8万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn