0

我想在 javascript 中使用 tensorflow 模型进行实时对象检测。为此,我正在关注 tensorflow.org 的本教程:https ://www.tensorflow.org/js/tutorials/conversion/import_keras 我在 keras 中训练了模型,将模型导出为 .h5 格式,将其转换为json 文件,最后导入模型。

这是我的 javascript 代码块的片段,

net = await tf.loadLayersModel('http://localhost:8000/converted/model.json');
const imgEl = document.getElementById('img');
const t = tf.browser.fromPixels(imgEl)
let result = await net.predict(t)

我要识别的图像是 255 x 255 ,RGB 我收到错误,输入图像是 3 维但预期是 4。我附上了错误的片段。 javascript中的错误

我尝试重新塑造输入图像但无济于事。有人可以帮忙吗?

4

0 回答 0