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();