0

0

如何利用机器学习库(如TensorFlow.js)在浏览器中运行AI模型?

紅蓮之龍

紅蓮之龍

发布时间:2025-10-09 14:17:01

|

288人浏览过

|

来源于php中文网

原创

答案:在浏览器中运行AI模型需将模型转为TensorFlow.js格式,通过异步加载、输入预处理和predict推理实现,结合WebGL加速与内存优化提升性能。

如何利用机器学习库(如tensorflow.js)在浏览器中运行ai模型?

浏览器中运行AI模型已成为前端智能化的重要方向,借助TensorFlow.js这类机器学习库,开发者可以直接在网页中加载和执行训练好的模型,无需依赖服务器端计算。整个过程不依赖后端,响应更快,也更保护用户隐私。

选择并准备合适的模型

要让AI模型在浏览器中运行,首先要确保模型是为Web环境优化过的。通常使用TensorFlow训练的模型需要转换成TensorFlow.js支持的格式(如.json和二进制权重文件)。

  • 使用tfjs-converter工具将Python中训练好的Keras或SavedModel模型转为Web可用格式。
  • 推荐使用轻量级模型(如MobileNet、Tiny YOLO),避免因体积过大影响页面加载速度。
  • 也可以直接使用TensorFlow.js提供的预训练模型,例如图像分类、姿态检测、语音识别等。

在网页中引入并加载模型

通过npm安装或CDN引入TensorFlow.js库后,即可在JavaScript中加载模型并准备推理。

Yodayo
Yodayo

一个专为动漫迷和vTuber打造的AI艺术创作平台、交流社区

下载
  • 使用tf.loadLayersModel()从远程或本地路径加载模型JSON文件。
  • 确保服务器支持CORS,否则跨域请求会失败。
  • 模型加载是异步操作,需用async/await或Promise处理完成后的逻辑。
示例代码:
const model = await tf.loadLayersModel('https://example.com/model.json');
console.log('模型加载完成');

预处理输入并执行推理

浏览器中的数据(如图像、音频)需要转换为张量(tensor)才能送入模型。

  • 使用tf.browser.fromPixels()将图像元素转为tensor,并调整尺寸和归一化。
  • 对输入进行批处理(batching)以匹配模型期望的shape。
  • 调用model.predict(inputTensor)获取输出结果。
图像分类示例:
const img = document.getElementById('inputImage');
const tensor = tf.browser.fromPixels(img)
  .resizeNearestNeighbor([224, 224])
  .toFloat()
  .div(tf.scalar(255))
  .expandDims();
  
const prediction = model.predict(tensor);

优化性能与用户体验

在浏览器中运行AI模型可能面临性能瓶颈,尤其在低端设备上。

  • 启用WebGL后端提升计算速度:tf.setBackend('webgl')
  • 避免频繁创建和销毁张量,使用tidy()清理中间变量防止内存泄漏。
  • 添加加载动画或进度提示,提升用户等待体验。
基本上就这些。只要模型适配得当,前端代码处理好数据流和资源管理,在浏览器中运行AI并不复杂,但细节容易忽略。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

WorkBuddy
WorkBuddy

腾讯云推出的AI原生桌面智能体工作台

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

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

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

457

2023.08.07

json是什么
json是什么

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

549

2023.08.23

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

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

337

2023.10.13

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

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

82

2025.09.10

js正则表达式
js正则表达式

php中文网为大家提供各种js正则表达式语法大全以及各种js正则表达式使用的方法,还有更多js正则表达式的相关文章、相关下载、相关课程,供大家免费下载体验。

531

2023.06.20

js获取当前时间
js获取当前时间

JS全称JavaScript,是一种具有函数优先的轻量级,解释型或即时编译型的编程语言;它是一种属于网络的高级脚本语言,主要用于Web,常用来为网页添加各式各样的动态功能。js怎么获取当前时间呢?php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

576

2023.07.28

js 字符串转数组
js 字符串转数组

js字符串转数组的方法:1、使用“split()”方法;2、使用“Array.from()”方法;3、使用for循环遍历;4、使用“Array.split()”方法。本专题为大家提供js字符串转数组的相关的文章、下载、课程内容,供大家免费下载体验。

760

2023.08.03

js是什么意思
js是什么意思

JS是JavaScript的缩写,它是一种广泛应用于网页开发的脚本语言。JavaScript是一种解释性的、基于对象和事件驱动的编程语言,通常用于为网页增加交互性和动态性。它可以在网页上实现复杂的功能和效果,如表单验证、页面元素操作、动画效果、数据交互等。

6231

2023.08.17

Python异步编程与Asyncio高并发应用实践
Python异步编程与Asyncio高并发应用实践

本专题围绕 Python 异步编程模型展开,深入讲解 Asyncio 框架的核心原理与应用实践。内容包括事件循环机制、协程任务调度、异步 IO 处理以及并发任务管理策略。通过构建高并发网络请求与异步数据处理案例,帮助开发者掌握 Python 在高并发场景中的高效开发方法,并提升系统资源利用率与整体运行性能。

37

2026.03.12

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
React 教程
React 教程

共58课时 | 6万人学习

TypeScript 教程
TypeScript 教程

共19课时 | 3.4万人学习

Bootstrap 5教程
Bootstrap 5教程

共46课时 | 3.6万人学习

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号