我如何加载tf js模型?

Question

我想加载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文件使用json-server，只是加载tfjs模型，但我得到这个错误

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

Uncaught (in promise) RangeError: Float32Array的字节长度应该是4的倍数

如何解决这个错误?

Answer 1

LoadModel在底层使用fetch。并且fetch不能直接访问本地文件。它是用来获取由服务器提供服务的文件。更多信息请点击这里。要用浏览器加载一个本地文件，有两种方法，要求用户使用或由服务器提供该文件。