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

Мы обучили модель 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)

• Затем я использовал механизм вывода в 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 

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

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

Спасибо,