Необработанное отклонение (ошибка): форма dict ['image_tensor'], предоставленная в model.execute (dict), должна быть [X, X, X, X], но должна быть X, X, X] - PullRequest
Новая
       18

Необработанное отклонение (ошибка): форма dict ['image_tensor'], предоставленная в model.execute (dict), должна быть [X, X, X, X], но должна быть X, X, X]

0 голосов
/ 17 февраля 2020

TensorFlow. js версия v1.5.2

Версия браузера Chrome 79.0.3945.130

Опишите проблему или запрос функции Я обучил свой собственный объект с помощью tenorflow 1.15.0 с fast_rcnn_inception_v2_coco_2018_01_28. После этого сохраненная модель была конвертирована веб-моделью с тензорным преобразователем. тензор потока js версия 1.5.2.

Эта созданная модель с использованием ошибки веб-приложения Форма dict ['image_tensor'], представленная в model.execute (dict), должна быть [X, X, X, X ], но был [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

тензор потока js веб-приложение 

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

Что плохого в этом процессе?

1 Ответ

0 голосов
/ 18 февраля 2020

Это ошибка несоответствия формы.

Кажется, что модель ожидает 4-мерный тензор, тогда как вы предоставляете 3D-тензор. expandDims или reshape может использоваться для добавления нового измерения к тензору 3d, используемому в качестве аргумента

...