1

试图让自定义 tensorFlowJS 模型在 JS 中工作但形状出错,请注意该模型已经过训练并转换为 model.json 但低于错误

错误:错误:concat4D 中的错误:张量 [1] (1,30,40,256) 的形状与沿非连接轴 1 的其余部分 (1,15,20,832) 的形状不匹配。

下面是代码:

const RGB = await imageToRgbaMatrix(imageUrl);
const model = await tf.loadLayersModel(modelUrl);
const ImageData =  await tf.tensor([RGB]);
const predictions = model.predict(ImageData);

简而言之,我正在尝试使用二进制文件实现 tfjs 模型(model.json),它给出:

concat4D 中的错误:张量 [1] (1,30,40,256) 的形状与其余 (1,15,20,832) 的形状不匹配。

4

0 回答 0