我有一个Tensorflow.js的model.json文件和相关的bin文件放在文件夹里,并且我已经创建了一个html文件。我希望在html文件中测试我的预测并获得反馈,但当我在服务器上运行它时却不工作,有什么想法吗?
<html><head><script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@latest"></script><script> async function run(){ const MODEL_URL = 'http://127.0.0.1:8887/model.json'; const model = await tf.loadLayersModel(MODEL_URL); console.log(model.summary()); const result = model.predict(input); alert(result); } run();</script><body></body></head>
在开发者工具中显示如下信息:
DevTools failed to load SourceMap: Could not load content for https://cdn.jsdelivr.net/npm/@tensorflow/tf.min.js.map: HTTP error: status code 404, net::ERR_HTTP_RESPONSE_CODE_FAILURE
它还显示配置格式不正确:
tfjs@latest:17 Uncaught (in promise) Error: layer: Improper config format: {"node":[{"name":"eedde323-f0a1-4978-8c57-3ae553bfb873/tf_op_layer_3ff186c1-fe28-4442-abe1-44f7ed005781/19c66cd4-e1d6-49f4-80ab-3aa348c5b264/ff9b2ff4-4799-495e-8e91-a6fed91e9315/GatherV2/3ff186c1-fe28-4442-abe1-44f7ed005781/19c66cd4-e1d6-49f4-80ab-3aa348c5b264/ff9b2ff4-4799-495e-8e91-a6fed91e9315/GatherV2/params","op":"Const","attr":{"value":{"tensor":{"dtype":"DTSTRING","tensorShape":{"dim":[{"size":"2"}]}}},"dtype":{"type":"DTSTRING"}}},{"name":"conv5_block1_0_conv/kernel","op":"Const","attr":{"value":{"tensor":{"dtype":"DTFLOAT","tensorShape":{"dim":[{"size":"1"},{"size":"1"},{"size":"1024"},{"size":"2048"}]}}},"dtype":{"type":"DTFLOAT"}}},{"name":"conv5_block1_0_conv/bias","op":"Const","attr":{"value":{"tensor":{"dtype":"DTFLOAT","tensorShape":{"dim":[{"size":"2048"}]}}},"dtype":{"type":"DTFLOAT"}}},{"name":"conv4_block1_0_conv/kernel","op":"Const","attr":{"dtype":{"type":"DTFLOAT"},"value":{"tensor":{"dtype":"DTFLOAT","tensorShape":{"dim":[{"size":"1"},{"size":"1"},{"size":"512"},{"size":"1024"}]}}}}},{"name":"conv4_block1_0_conv/bias","op":"Const","attr":{"dtype":{"type":"DTFLOAT"},"value":{"tensor":{"dtype":"DTFLOAT","tensorShape":{"dim":[{"size":"1024"}]}}}}},{"name":"conv3_block1_0_conv/kernel","op":"Const","attr":{"dtype":{"type":"DTFLOAT"},"value":{"tensor":{"dtype":"DTFLOAT","tensorShape":{"dim":[{"size":"1"},{"size":"1"},{"size":"256"},{"size":"512"}]}}}}},{"name":"conv3_block1_0_conv/bias","op":"Const","attr":{"dtype":{"type":"DTFLOAT"},"value":{"tensor":{"dtype":"DTFLOAT","tensorShape":{"dim":[{"size":"512"}]}}}}},{"name":"conv2_block1_0_conv/kernel","op":"Const","attr":{"dtype":{"type":"DTFLOAT"},"value":{"tensor":{"dtype":"DTFLOAT","tensorShape":{"dim":[{"size":"1"},{"size":"1"},{"size":"64"},{"size":"256"}]}}}}},{"name":"conv2_block1_0_conv/bias","op":"Const","attr":{"dtype":{"type":"DTFLOAT"},"value":{"tensor":{"dtype":"DTFLOAT","tensorShape":{"dim":[{"size":"256"}]}}}}},{"name":"Image","op":"Placeholder","attr":{"dtype":{"type":"DTFLOAT"},"shape":{"shape":{"dim":[{"size":"-1"},{"size":"224"},{"size":"224"},{"size":"3"}]}}}},{"name":"saved_resnet_50_model/tf_op_layer_Mul/Mul/y","op":"Const","attr":{"value":{"tensor":{"dtype":"DTFLOAT","tensorShape":{}}},"dtype":{"type":"DTFLOAT"}}},{"name":"saved_resnet_50_model/tf_op_layer_Sub/Sub/y","op":"Const","attr":{"dtype":{"type":"DTFLOAT"},"value":{"tensor":{"dtype":"DTFLOAT","tensorShape":{}}}}},{"name":"saved_resnet_50_model/resnet50v2/conv1_pad/Pad/paddings","op":"Const","attr":{"value":{"tensor":{"dtype":"DTINT32","tensorShape":{"dim":[{"size":"4"},{"size":"2"}]}}},"dtype":{"type":"DTINT32"}}},{"name":"conv1_conv/kernel","op":"Const","attr":{"dtype":{"type":"DTFLOAT"},"value":{"tensor":{"dtype":"DTFLOAT","tensorShape":{"dim":[{"size":"7"},{"size":"7"},{"size":"3"},{"size":"64"}]}}}}},{"name":"conv1_conv/bias","op":"Const","attr":{"dtype":{"type":"DTFLOAT"},"value":{"tensor":{"dtype":"DTFLOAT","tensorShape":{"dim":[{"size":"64"}]}}}}},{"name":"saved_resnet_50_model/resnet50v2/pool1_pad/Pad/paddings","op":"Const","attr":{"value":{"tensor":{"dtype":"DTINT32","tensorShape":{"dim":[{"size":"4"},{"size":"2"}]}}},"dtype":{"type":"DTINT32"}}},{"name":"saved_resnet_50_model/resnet50v2/conv2_block1_2_pad/Pad/paddings","op":"Const","attr":{"dtype":{"type":"DTINT32"},"value":{"tensor":{"dtype":"DTINT32","tensorShape":{"dim":[{"size":"4"},{"size":"2"}]}}}}},{"name":"conv2_block1_3_conv/kernel","op":"Const","attr":{"dtype":{"type":"DTFLOAT"},"value":{"tensor":{"dtype":"DTFLOAT","tensorShape":{"dim":[{"size":"1"},{"size":"1"},{"size":"64"},{"size":"256"}]}}}}},{"name":"conv2_block1_3_conv/bias","op":"Const","attr":{"value":{"tensor":{"dtype":"DTFLOAT","tensorShape":{"dim":[{"size":"256"}]}}},"dtype":{"type":"DTFLOAT"}}},{"name":"saved_resnet_50_model/resnet50v2/conv2_block2_2_pad/Pad/paddings","op":"Const","attr":{"dtype":{"type":"DTINT32"},"value":{"tensor":{"dtype":"DTINT32","tensorShape":{"dim":[{"size":"4"},{"size":"2"}]}}}}},{"name":"conv2_block2_3_conv/kernel","op":"Const","attr":{"value":{"tensor":{"dtype":"DTFLOAT","tensorShape":{"dim":[{"size":"1"},{"size":"1"},{"size":"64"},{"size":"256"}]}}},"dtype":{"type":"DTFLOAT"}}},{"name":"conv2_block2_3_conv/bias","op":"Const","attr":{"value":{"tensor":{"dtype":"DTFLOAT","tensorShape":{"dim":[{"size":"256"}]}}},"dtype":{"type":"DTFLOAT"}}},{"name":"saved_resnet_50_model/resnet50v2/conv2_block3_2_pad/Pad/paddings","op":"Const","attr":{"value":{"tensor":{"dtype":"DTINT32","tensorShape":{"dim":[{"size":"4"},{"size":"2"}]}}},"dtype":{"type":"DTINT32"}}},{"name":"conv2_block3_3_conv/kernel","op":"Const","attr":{"value":{"tensor":{"dtype":"DTFLOAT","tensorShape":{"dim":[{"size":"1"},{"size":"1"},{"size":"64"},{"size":"256"}]}}},"dtype":{"type":"DTFLOAT"}}},{"name":"conv2_block3_3_conv/bias","op":"Const","attr":{"value":{"tensor":{"dtype":"DTFLOAT","tensorShape":{"dim":[{"size":"256"}]}}},"dtype":{"type":"DTFLOAT"}}},{"name":"saved_resnet_50_model/resnet50v2/conv3_block1_2_patfjs@latest:17) at tA (tfjs@latest:17) at tfjs@latest:17 at u (tfjs@latest:17) at Generator._invoke (tfjs@latest:17) at Generator.forEach.e.<computed> [as next] (tfjs@latest:17) at Um (tfjs@latest:17) at o (tfjs@latest:17)Jm @ tfjs@latest:17r @ tfjs@latest:17t @ tfjs@latest:17eE @ tfjs@latest:17tA @ tfjs@latest:17(anonymous) @ tfjs@latest:17u @ tfjs@latest:17(anonymous) @ tfjs@latest:17forEach.e.<computed> @ tfjs@latest:17Um @ tfjs@latest:17o @ tfjs@latest:17
回答:
我可以通过以下步骤避免模型的错误:
-
以
.h5
格式保存Tensorflow模型:model.save('sampleModel.h5')
-
将
.h5
文件转换为tfjs_graph_model
格式以便在tensorflow.js中使用:tensorflowjs_converter --input_format keras --output_format tfjs_graph_model sampleModel.h5 sampleModel
-
使用
loadGraphModel()
而不是loadLayerModel()
来加载模型:<html><head><script src="https://cdn.jsdelivr.net/npm/@tensorflow/[email protected]/dist/tf.min.js"></script></head><body><script>tf.loadGraphModel('sampleModel/model.json')</script></body></html>
欲了解更多信息,请查看这里
为了避免导入tensorflow.js库时出现错误,请使用此链接:"https://cdn.jsdelivr.net/npm/@tensorflow/tfjs"
,而不是此链接:"https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@latest"
。