tensorflow 如何加载tf js模型?

aor9mmx1  于 2022-11-16  发布在  其他
关注(0)|答案(1)|浏览(295)

我想加载tf js模型(从keras h5转换),但模型未打开
我试着训练keras模型并将其转换为tfjs并使用它
我的Keras模型如下

pre_model = keras.applications.mobilenet_v2.MobileNetV2(include_top=False, input_shape=[224, 224, 3], weigths="imagenet")
input1 = keras.layers.Input(shape=(224, 224, 3), dtype=tf.float32)
input2 = keras.layers.Input(shape=(106,), dtype=tf.float32)
feat = pre_model(input1)
#fully connected layers
#x1 is tensor of input1, x2 is tensor of input2
x = keras.layers.concatenate([x1, x2])
#fully connected layers
output = keras.layers.Dense(5, activation="softmax")
model = keras.models.Model(inputs=[input1, input2], outputs=[output])

我使用json-server提供model.json文件,并且只加载tfjs模型,但是我得到了这个错误

async function loadmodel() {
    const model = await tf.loadLayersModel('http://localhost:4000/file');
}
loadmodel()

未捕获(在承诺中)范围错误:Float 32Array的字节长度应为4的倍数
如何解决此错误?

wj8zmpe1

wj8zmpe11#

LoadModel使用了fetch。fetch不能直接访问本地文件。它是用来获取服务器提供的文件的。这里有更多关于这方面的信息。要用浏览器加载本地文件,有两种方法,要求用户上传文件或由服务器提供文件。

相关问题