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.

OpenVINO and Detectron

Shevchuk__Mikhailo
750 Views
0 Kudos
3 Replies
Shubha_R_Intel
Employee
750 Views

Dear Shevchuk, Mikhailo:

The document below tells you exactly how to generate IR for maskrcnn_r50_fpn_2x as described:

https://docs.openvinotoolkit.org/latest/_docs_MO_DG_prepare_model_convert_model_tf_specific_Convert_Object_Detection_API_Models.html

Please select your Mask RCNN model from one of OpenVino's validated models here:

https://docs.openvinotoolkit.org/latest/_docs_MO_DG_prepare_model_convert_model_Convert_Model_From_TensorFlow.html

Thanks !

Shubha

 

0 Kudos
Juj__Satyajith
Beginner
750 Views

Dear Shubha,

That specific example is for a tensorflow model, right? If I remember correctly, model optimisation does not currently support the Detectron framework.

Correct me if I am wrong.

0 Kudos
Shubha_R_Intel
Employee
750 Views

Dear folks, This OpenVino documentation definitely mentions Detectron, so it is supported. The key though is to convert a *.pkl file to a *.pb first, since Model Optimizer does not support *.pkl.  There is a script in the Detectron github repo which enables conversion:

https://github.com/facebookresearch/Detectron/blob/master/tools/convert_pkl_to_pb.py

Thanks,

Shubha

0 Kudos
Reply