我在 Keras 中创建了一个模型并将其转换为 Tensorflow.js 模型并将其加载到我的 node.js 项目中。现在我想从 Tensorflow.js 中的这个模型中得到预测。我已经弄清楚如何加载单个图像:
var singleImageData = fs.readFileSync('path/to/image.jpeg');
var image = tf.node.decodeImage(new Uint8Array(singleImageData), 3);
image = tf.cast(image, 'float32');
//other image processing
这将创建一个 shape 的张量(imageWidth, imageHeight, 3)
。但我想将一批图像加载到一个 shape 的张量中(batchNumber, imageWidth, imageHeight, 3)
。
我该怎么做呢?