0

TensorFlow.js 版本 v1.5.2

浏览器版本 Chrome 79.0.3945.130

描述我使用 tensorflow 1.15.0 和 faster_rcnn_inception_v2_coco_2018_01_28 训练自己的对象的问题或功能请求。之后保存的模型使用 tensorflow 转换器转换 web 模型。tensorflowjs 版本是 1.5.2。

This created model using web app error The shape of dict['image_tensor'] provided in model.execute(dict) must be [X,X,X,X], but was [X,X,X]

创建保存的模型

python export_inference_graph.py --input_type image_tensor --pipeline_config_path training/faster_rcnn_inception_v2_pets.config --trained_checkpoint_prefix training/model.ckpt-XXXX --output_directory inference_graph

转换保存的模型

tensorflowjs_converter
--input_format=tf_saved_model
--output_node_names='MobilenetV1/Predictions/Reshape_1'
--saved_model_tags=serve
/mobilenet/saved_model
/mobilenet/web_model

tensorflowjs 网络应用

const img = document.getElementById('img');
const model = await tf.loadGraphModel('model.json')
model.predict(tf.browser.fromPixels(img)) // or model.predict(img)

这个过程有什么问题?

4

2 回答 2

0

可能会出现此错误,因为模型需要一批图像而不是只有一个图像。要修复它,可以使用expandDims.

const output = model.predict(tf.fromPixels(image).expandDims(0));
于 2021-04-03T15:47:48.537 回答
0

这是形状不匹配错误。

看起来模型需要一个 4d 张量,而您提供的是 3d 张量。expandDimsreshape可用于向用作参数的 3d 张量添加新维度

于 2020-02-18T12:40:18.493 回答