跳到主要内容

tf.loadGraphModel

2025年02月11日
柏拉文
越努力,越幸运

一、认识


tf.loadGraphModel(modelUrl, options?, tfio?) 用于从指定 URL 加载计算图(Graph Model)模型。该函数会根据提供的 modelUrl 加载一个计算图模型(Graph Model),并返回一个 tf.GraphModel 实例。计算图模型通常是 TensorFlow.js 转换自 TensorFlow(Python) 训练的模型,并以 model.json*.bin 格式存储。

二、语法


tf.loadGraphModel (modelUrl, options?, tfio?)

三、用法


const modelUrl =
'https://storage.googleapis.com/tfjs-models/savedmodel/mobilenet_v2_1.0_224/model.json';
const model = await tf.loadGraphModel(modelUrl);
const zeros = tf.zeros([1, 224, 224, 3]);
model.predict(zeros).print();