Intel® Distribution of OpenVINO™ Toolkit
Community assistance about the Intel® Distribution of OpenVINO™ toolkit, OpenCV, and all aspects of computer vision-related on Intel® platforms.

Convert torch vision maskrcnn to IR format

Abdulsathar
Beginner
600 Views

Hi,

Is it possible to convert a trained torch vision maskrcnn model to OpenVINO model. I am encountering different errors. if not, which maskrcnn is supported by OpenVINO currently?

 

0 Kudos
2 Replies
IntelSupport
Community Manager
577 Views

Hi Abdul Sathar,

 

Thanks for reaching out.

 

Pytorch framework is supported in OpenVINO by exporting the model into ONNX format. Refer Converting a PyTorch* Model for more details on the conversion.

 

Apart from that, you can find the list of Mask R-CNN models from OpenVINO™ Toolkit Public Pre-Trained Models.

 

Below is the example of converting Mask R-CNN Inception ResNet V2 1024x1024 from TensorFlow 2 Detection Model Zoo. Here is the command that I used:

 

mo --saved_model_dir <PATH_TO_SAVED_MODEL>\saved_model --transformations_config <INSTALL_DIR>\openvino\tools\mo\front\tf\mask_rcnn_support_api_v2.0.json --tensorflow_object_detection_api_pipeline_config <PATH_TO_PIPELINE_CONFIG>\pipeline.config --reverse_input_channels

 

mask_rcnn_pass (2).jpg

 

 

Regards,

Aznie

 

0 Kudos
IntelSupport
Community Manager
526 Views

Hi Abdul Sathar,


This thread will no longer be monitored since we have provided the information. If you need any additional information from Intel, please submit a new question.



Regards,

Aznie


0 Kudos
Reply