网易首页 > 网易号 > 正文 申请入驻

TensorFlow.js发布:使用JS进行机器学习并在浏览器中运行

0
分享至

TensorFlow官方团队近日发布多个重大更新,其中包括:为python开发者提供的,可以立即评估操作并且无需额外图形构建步骤的eager execution;可以促进机器学习模型可重用部分的发布、发现和使用的TensorFlow Hub;针对移动和嵌入式设备的轻量级解决方案TensorFlow Lite;Swift的TensorFlow开源;面向JavaScript开发者的新机器学习框架TensorFlow.js!

介绍

TensorFlow.js是为JavaScript开发者准备的开源库,可以使用JavaScript和高级图层API完全在浏览器中定义,训练和运行机器学习模型!如果你是一名机器学习新手,那么TensorFlow.js是开始学习的好方法。

浏览器内的机器学习

在浏览器中完全由客户端运行的机器学习程序将会解锁新的机会,如交互式机器学习!例如下方链接中的吃豆人游戏。

https://github.com/tensorflow/tfjs-examples

使用神经网络将你的摄像头变成PAC-MAN的控制器

如果你想尝试其他游戏,可以使用手机上的浏览器玩Emoji Scavenger Hunt。

https://emojiscavengerhunt.withgoogle.com/

项目:https://github.com/google/emoji-scavenger-hunt

从用户的角度来看,在浏览器中运行的ML意味着不需要安装任何库或驱动程序。只需打开网页,你的程序就可以运行了。此外,它已准备好使用GPU加速运行。TensorFlow.js自动支持WebGL,并在GPU可用时会加速代码。用户也可以通过移动设备打开你的网页,在这种情况下,模型可以利用传感器数据,例如陀螺仪或加速度传感器。最后,所有数据都保留在客户端上,使得TensorFlow.js可用于低延迟推理以及隐私保护程序。

你可以用TensorFlow.js做什么?

如果使用TensorFlow.js进行开发,可以考虑以下三种工作流程。

  • 你可以导入现有的预训练的模型进行推理。如果你有一个现成的TensorFlow或Keras模型,则可以将其转换为TensorFlow.js格式,并将其加载到浏览器中进行推理。
  • 你可以导入的模型进行再训练。就像上面的吃豆人演示一样,你可以使用迁移学习来增强现有预训练好的离线模型(使用在浏览器中收集的少量数据),使用的技术称为图像再训练(Image Retraining)。这是只使用少量数据,快速训练准确模型的一种方法。
  • 直接在浏览器中创作模型。你还可以使用TensorFlow.js,完全在浏览器中使用Javascript和高级层API定义,训练和运行模型。如果你熟悉Keras,那么高级层API应该也会很熟悉。

让我们看看代码

如果你愿意,可以直接到示例或教程开始。

示例:https://github.com/tensorflow/tfjs-examples

教程:http://js.tensorflow.org/

以下内容展示了如何在浏览器中导出用Python定义的模型进行推理,以及如何完全用Javascript定义和训练模型。这是定义一个神经网络来对花朵进行分类的代码片段,就像在TensorFlow.org的入门指南中一样(也就是说分类鸢尾花)。在这里我使用一堆层定义一个模型。

1import * as tffrom ‘@tensorflow/tfjs’;

2const model= tf.sequential();

3model.add(tf.layers.dense({inputShape: [4], units:100}));

4model.add(tf.layers.dense({units:4}));

5model.compile({loss: ‘categoricalCrossentropy’, optimizer: ‘sgd’});

我们在此使用的层API支持在示例目录中能找到的所有Keras层(包括Dense,CNN,LSTM等)。然后,我们可以使用与方法调用相同的 Keras-compatible API来训练我们的模型:

1await model.fit(

2 xData, yData, {

3 batchSize: batchSize,

4 epochs: epochs

5});

这个模型现在可以用来做预测了:

01// Get measurementsfor a new flower to generate a prediction

02// The first argumentis the data,and the secondis the shape.

03const inputData= tf.tensor2d([[4.8,3.0,1.4,0.1]], [1,4]);

04

05// Get the highest confidence predictionfrom our model

06const result= model.predict(inputData);

07const winner= irisClasses[result.argMax().dataSync()[0]];

08

09// Display the winner

10console.log(winner);

TensorFlow.js还包含底层API(以前称为deeplearn.js)并支持Eager execution。

TensorFlow.js API

TensorFlow.js如何与deeplearn.js关联?

TensorFlow.js属于JavaScript生态系统,实现其进行机器学习的工具,继承自deeplearn.js(即TensorFlow.js Core)。TensorFlow.js还包含一个Layers API,它是使用Core构建机器学习模型的高级库,并且也是用于自动移植TensorFlow SavedModels和Keras hdf5模型的工具。

本文为编译作品,转载请注明出处。

特别声明:以上内容(如有图片或视频亦包括在内)为自媒体平台“网易号”用户上传并发布,本平台仅提供信息存储服务。

Notice: The content above (including the pictures and videos if any) is uploaded and posted by a user of NetEase Hao, which is a social media platform and only provides information storage services.

相关推荐
热点推荐
哈尔滨冰雪大世界发布闭园公告:冰雪景观融化严重,退票游客可免费游览下一届

哈尔滨冰雪大世界发布闭园公告:冰雪景观融化严重,退票游客可免费游览下一届

环球网资讯
2026-02-22 15:24:25
15年前,一位高手教给我15条“老掉牙”的交易规则,却让人年年赚钱

15年前,一位高手教给我15条“老掉牙”的交易规则,却让人年年赚钱

汇商Forexpress
2026-02-03 10:00:00
告别“例行公事”,这几个技巧让你高潮不断

告别“例行公事”,这几个技巧让你高潮不断

精彩分享快乐
2026-01-08 19:18:46
“中国要被我们拖垮了!”24年,韩国掀起拒绝吃中国进口白菜浪潮

“中国要被我们拖垮了!”24年,韩国掀起拒绝吃中国进口白菜浪潮

扶苏聊历史
2026-01-23 10:28:26
比福建舰还大!4艘万吨巨船同时出坞,美这才明白,输给中国了

比福建舰还大!4艘万吨巨船同时出坞,美这才明白,输给中国了

温读史
2026-01-09 23:42:22
哈佛大学:被父母用心关注的孩子,其大脑发育速度比同龄人快30%

哈佛大学:被父母用心关注的孩子,其大脑发育速度比同龄人快30%

追寻文史
2026-02-21 22:24:30
5连胜!胜率超雷霆,联盟第1!本赛季头号黑马,剑指队史第4冠

5连胜!胜率超雷霆,联盟第1!本赛季头号黑马,剑指队史第4冠

弄月公子
2026-02-22 13:01:35
刘涛意外登上热搜,与妈祖换人事件对比鲜明,敬畏之心不可丢

刘涛意外登上热搜,与妈祖换人事件对比鲜明,敬畏之心不可丢

然哥闲聊
2026-02-22 09:48:32
早上七点起床错了?医生建议:过了70岁,起床要尽量做到这7点

早上七点起床错了?医生建议:过了70岁,起床要尽量做到这7点

泠泠说史
2025-10-23 16:41:51
欧文复出疑云再起!独行侠删训练视频,联盟摆烂新规藏猫腻!

欧文复出疑云再起!独行侠删训练视频,联盟摆烂新规藏猫腻!

篮球小布丁
2026-02-23 03:36:33
深夜十点!宏远陈老板约谈杜锋,徐杰离开国家队,贝兹利官宣回归

深夜十点!宏远陈老板约谈杜锋,徐杰离开国家队,贝兹利官宣回归

多特体育说
2026-02-22 21:52:55
4年财务造假,股价从最高42元跌到0.29元,最后退市

4年财务造假,股价从最高42元跌到0.29元,最后退市

财经智多星
2026-02-22 21:35:21
中国游客在新西兰又出事!护照上交,还要花钱消灾,“有些东西,钱都赔不了”

中国游客在新西兰又出事!护照上交,还要花钱消灾,“有些东西,钱都赔不了”

发现新西兰
2026-02-20 11:48:23
4-1,阿森纳终结2轮英超不胜+5分优势领跑,埃泽、哲凯赖什双响

4-1,阿森纳终结2轮英超不胜+5分优势领跑,埃泽、哲凯赖什双响

侧身凌空斩
2026-02-23 02:37:12
太敢说!咸鱼被称国内黑市,果然够全面

太敢说!咸鱼被称国内黑市,果然够全面

另子维爱读史
2026-02-03 22:10:36
巴恩斯利青训主管:徐彬态度极佳,是位令人难以置信的运动员

巴恩斯利青训主管:徐彬态度极佳,是位令人难以置信的运动员

懂球帝
2026-02-22 15:24:29
火箭队主帅在惨败尼克斯后,公开批评阿门·汤普森和里德·谢泼德

火箭队主帅在惨败尼克斯后,公开批评阿门·汤普森和里德·谢泼德

好火子
2026-02-22 22:51:45
“初中女生被老人索赔22万”,家长最新回应:对方已撤诉,不希望再占用公共资源

“初中女生被老人索赔22万”,家长最新回应:对方已撤诉,不希望再占用公共资源

封面新闻
2026-02-22 00:27:04
报复一个人最狠的方法,不是翻脸,不是打压,而是佛家三句话

报复一个人最狠的方法,不是翻脸,不是打压,而是佛家三句话

瓜哥的动物日记
2026-02-14 00:59:40
1-0!1.1亿先生临阵伤退,麦卡96分钟绝杀,利物浦客场复仇

1-0!1.1亿先生临阵伤退,麦卡96分钟绝杀,利物浦客场复仇

我的护球最独特
2026-02-23 00:03:56
2026-02-23 04:28:49
AiTechYun
AiTechYun
专注于人工智能
101文章数 79关注度
往期回顾 全部

科技要闻

马斯克:星舰每年将发射超过10000颗卫星

头条要闻

男子持霰弹枪燃烧罐闯特朗普私宅被击毙 细节披露

头条要闻

男子持霰弹枪燃烧罐闯特朗普私宅被击毙 细节披露

体育要闻

谷爱凌:6次参赛6次夺牌 我对自己非常自豪

娱乐要闻

谷爱凌:真正的强大 敢接纳生命的节奏

财经要闻

特朗普新加征关税税率从10%提升至15%

汽车要闻

续航1810km!smart精灵#6 EHD超级电混2026年上市

态度原创

数码
游戏
健康
艺术
房产

数码要闻

古尔曼:苹果3月2 - 4日发布“至少五款产品”

《GTA6》的第二天发售?《宝可梦》新作爆料来了!

转头就晕的耳石症,能开车上班吗?

艺术要闻

谁能想到,“饺子包”火了!还是韭菜鸡蛋味儿,超吸睛!

房产要闻

窗前即地标!独占三亚湾C位 自贸港总裁行宫亮相

无障碍浏览 进入关怀版