Closed laitathei closed 1 year ago
Hi @laitathei , we apologize for the delay, I just pinged the ML team for assistance here.
Hey @laitathei , we are investigating. In the meantime could you try the code from this comment in #353. It was tested with:
torch==1.9.1
torchvision==0.10.0
openvino-dev==2021.4.2
Hi @laitathei , based on @tersekmatija 's comment and by investigating mo_pytorch
, we've found a working solution for exporting Mask R-CNN model with openvino==2021.4.2
. Here is a notebook that produces a blob which you can use in the mentioned example.
@klemen1999 thank you for your helping. The notebook is exactly what i want. It can convert the mask rcnn model to blob format very well !
I have followed https://github.com/openvinotoolkit/openvino_contrib/tree/master/modules/mo_pytorch guide line to convert model. it seems no problem with this guideline. Alexnet model can successfully convert but maskrcnn_resnet50_fpn cannot be able to convert.
The following code is what i have done:
Then, it generate following error
It is good for providing tutorial of https://github.com/luxonis/depthai-experiments/tree/master/gen2-maskrcnn-resnet50 for converting the model from pytorch to openvino. Thank you.