Вход или выход Softmax SecondStageBoxPredictor/ClassPredictor/BiasAdd/softmax имеет недопустимую партию

Question

Мы обучили Mask R-CNN модель на NVIDIA GPU выполнить сегментацию экземпляра объекта и протестировать на некоторых изображениях с достаточной производительностью Теперь мы собираемся развернуть обученную модель на Neural Compute Stick 2, Я только начинаю с OpenVINO toolkit и вот что я сделал:

• Я загрузил mask_rcnn_inception_v2_coco.tar.gz от TensorFlow detection model zoo и распаковал его.

• я использовал ModelOptimizer следующим образом, чтобы получить Intermediate Representation:

  python3 mo_tf.py \
  --input_model ./frozen_inference_graph.pb \
  -- tensorflow_use_custom_operations_config extensions/front/tf/mask_rcnn_support.json \
  --tensorflow_object_detection_api_pipeline_config ./pipeline.config \
  --data_type FP16
  

(Я использовал тип данных FP16, так как по умолчанию FP32 не поддерживается на VPU)

• Затем я использовал Inference Engine в mask_rcnn_demo следующее:

./mask_rcnn_demo -m ./frozen_graph.xml -i ./image.jpg -d MYRIADОднако я получил следующую ошибку:

[ ERROR ] [VPU] Softmax input or output 
SecondStageBoxPredictor/ClassPredictor/BiasAdd/softmax has invalid batch 

Может ли кто-нибудь указать мне источник этой ошибки?

Из документации я понимаю, что в настоящее время маска RCNN поддерживается только на процессорах и графических процессорах, но я хотел бы знать, что я могу сделать, чтобы запустить его на VPU (например, написать пользовательские слои для слоев, не поддерживаемых в оптимизаторе моделей?), Я не нашел объяснения того, почему маска RCNN не поддерживается в VPU в документации.

Спасибо,

Answer 1

Попробовать с плагином HETERO

-d ГЕТЕРО: МИРИАДА, ЦП