0

0

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

紅蓮之龍

紅蓮之龍

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

|

641人浏览过

|

来源于php中文网

原创

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

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

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

1. 引入 TensorFlow.js

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

引入后,你就可以在 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');

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

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

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

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

玻璃钢企业网站源码1.5
玻璃钢企业网站源码1.5

本程序源码为asp与acc编写,并没有花哨的界面与繁琐的功能,维护简单方便,只要你有一些点点asp的基础,二次开发易如反掌。 1.功能包括产品,新闻,留言簿,招聘,下载,...是大部分中小型的企业建站的首选。本程序是免费开源,只为大家学习之用。如果用于商业,版权问题概不负责。1.采用asp+access更加适合中小企业的网站模式。 2.网站页面div+css兼容目前所有主流浏览器,ie6+,Ch

下载
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工具

更多
DeepSeek
DeepSeek

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

豆包大模型
豆包大模型

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

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

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

文心一言
文心一言

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

讯飞写作
讯飞写作

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

即梦AI
即梦AI

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

ChatGPT
ChatGPT

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

相关专题

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

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

420

2023.08.07

json是什么
json是什么

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

536

2023.08.23

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

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

311

2023.10.13

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

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

77

2025.09.10

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

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

531

2023.09.20

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

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

81

2025.09.18

python 全局变量
python 全局变量

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

96

2025.09.18

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

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

515

2023.06.20

C++ 设计模式与软件架构
C++ 设计模式与软件架构

本专题深入讲解 C++ 中的常见设计模式与架构优化,包括单例模式、工厂模式、观察者模式、策略模式、命令模式等,结合实际案例展示如何在 C++ 项目中应用这些模式提升代码可维护性与扩展性。通过案例分析,帮助开发者掌握 如何运用设计模式构建高质量的软件架构,提升系统的灵活性与可扩展性。

14

2026.01.30

热门下载

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

精品课程

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

共58课时 | 4.4万人学习

TypeScript 教程
TypeScript 教程

共19课时 | 2.6万人学习

Bootstrap 5教程
Bootstrap 5教程

共46课时 | 3.1万人学习

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

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