javascript如何实现机器学习_TensorFlow.js能运行哪些模型？-js教程-PHP中文网

javascript如何实现机器学习_TensorFlow.js能运行哪些模型？

幻影之瞳

发布： 2025-12-21 05:23:26

原创

122人浏览过

TensorFlow.js支持三类模型：预训练模型（如cocossd、blazeface）、Python训练后转换的自定义模型、纯前端小规模训练模型；需注意加载性能、兼容性、隐私及终端适配。

javascript如何实现机器学习_tensorflow.js能运行哪些模型？

JavaScript 本身不内置机器学习能力，但通过 TensorFlow.js（简称 TF.js），你可以在浏览器或 Node.js 环境中直接运行训练好的模型、微调模型，甚至从零训练简单模型。

TensorFlow.js 能运行哪些模型？

TF.js 支持三类主流使用方式，对应不同复杂度和场景：

预训练模型（即开即用）：官方提供多个已转换、可直接加载的模型，比如：
• face-api.js（基于 TF.js 的人脸检测/识别库，非官方但广泛使用）
• cocossd（实时物体检测，支持 80 类 COCO 数据集）
• blazeface（轻量级人脸检测，适合移动端）
• handpose（手部关键点检测）
• speech-commands（关键词语音识别，如 “yes” / “no”）
• posenet（人体姿态估计）
自定义训练模型（Python → JS）：你在 Python 中用 TensorFlow/Keras 训练好模型后，导出为 SavedModel 或 Keras HDF5 格式，再用 tensorflowjs_converter 工具转成 TF.js 可加载的 JSON + 二进制权重格式。支持大多数常见结构（CNN、RNN、Transformer 小型变体等），但需注意：
• 不支持所有 TensorFlow 操作（如某些控制流、动态 shape 操作）
• 复杂模型（如大参数量 LLM）在浏览器中推理可能卡顿，建议裁剪或量化
纯前端训练（适合小规模任务）：TF.js 提供完整的张量操作与自动微分，可在浏览器中训练简单模型，例如：
• 线性回归 / 逻辑回归
• 小型全连接网络（如 MNIST 手写数字分类）
• 自编码器、GAN（实验性质，收敛慢）
• 注意：浏览器训练受 CPU/GPU 资源限制，不适用于大数据集或深层网络

一个最简运行示例（COCO-SSD）

几行代码即可完成物体检测：

Uni-CourseHelper

私人AI助教，高效学习工具

查看详情

import * as tf from '@tensorflow/tfjs';
import * as cocoSsd from '@tensorflow-models/coco-ssd';
<p>// 加载模型（首次会下载约 10MB）
const model = await cocoSsd.load();</p><p>// 对图像元素做推理
const img = document.getElementById('myImage');
const predictions = await model.detect(img);</p><p>// predictions 是包含 { bbox: [x,y,w,h], class: 'cat', score: 0.92 } 的数组
console.log(predictions);

登录后复制