Мы обучили модель 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 в документации.
Спасибо,