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

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）

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

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

关键注意事项

• 模型加载耗时且占内存：首次加载需网络请求 + 解析权重，建议加 loading 状态；大模型可考虑懒加载或服务端预推理
• 浏览器兼容性：TF.js 优先使用 WebGL 加速，部分老旧设备或 Safari 需回退到 CPU 模式（性能明显下降）
• 隐私与安全：所有数据保留在用户本地，适合敏感场景（如人脸分析不上传）；但也意味着无法利用云端算力
• Node.js 支持：通过 @tensorflow/tfjs-node 绑定原生 C++ 后端，性能接近 Python，适合服务端模型部署

不复杂但容易忽略：模型选择要匹配终端能力——移动端优先选 blazeface 而非 Posenet，Web 应用慎用未量化的 ResNet50。