Ошибка преобразования TensorFlow Mobil eNet в OpenVINO IR (ошибка конвертера моделей) - PullRequest
Новая
       96

Ошибка преобразования TensorFlow Mobil eNet в OpenVINO IR (ошибка конвертера моделей)

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

Я пытался преобразовать Mobil enet в OpenVINO, и я получил эту ошибку: 

[ ERROR ]  Cannot infer shapes or values for node "Preprocessor/map/TensorArrayStack/TensorArrayGatherV3".
[ ERROR ]
[ ERROR ]
[ ERROR ]  It can happen due to bug in custom shape infer function <function TensorArrayGather.array_infer at 0x000001A80D374510>.
[ ERROR ]  Or because the node inputs have incorrect values/shapes.
[ ERROR ]  Or because input shapes are incorrect (embedded to the model or passed via --input_shape).
[ ERROR ]  Run Model Optimizer with --log_level=DEBUG for more information.
[ ERROR ]
Stopped shape/value propagation at "Preprocessor/map/TensorArrayStack/TensorArrayGatherV3" node.

Вся информация о моей рабочей среде:

Используемая команда оптимизатора модели: 

C:\Program Files (x86)\IntelSWTools\openvino_2019.3.379>python deployment_tools\model_optimizer\mo_tf.py --input_model tensorflow_model\frozen_inference_graph_face.pb --input_shape (1,3,300,300) --output_dir vb_openvino_models

Версия OpenVINO: openvino_2019.3.379;

ОС: Windows 10;

DL Framework: TensorFlow;

Полный файл отладки журнала: Прикреплено .

Пб полной модели можно скачать здесь: https://drive.google.com/open?id=0B5ttP5kO_loUdWZWZVVrN2VmWFk

Слой / операция, которая выдает ошибку, называется TensorArrayGatherV3, я ее искал в поддерживаемых операциях OpenVINO для TensorFlow, и я обнаружил, что он поддерживается только тогда, когда он сливается со слоем TensorIterator (https://docs.openvinotoolkit.org/latest/_docs_MO_DG_prepare_model_Supported_Frameworks_Layers.html). Так что я думаю, что это проблема, потому что моя слита с TensorArrayStack (я думаю). Могу ли я изменить код / ​​модель TensorFlow так, чтобы TensorArrayStack отображался в слой TensorIterator?

...