Export Mask R-CNN to ONNX format

Question

Karl Mueller 2023-5-9

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1960099-export-mask-r-cnn-to-onnx-format

评论： Karl Mueller 2023-5-18

I am considering using MATLAB to train a Mask R-CNN network but the trained network must be able to be exported to ONNX format afterwards so I can import and use it in my application.

I took a look at the exportONNXFormat function and exporting a roiAlignLayer doesn't seem to be listed. Can anyone confirm whether this kind of layer is compatible with the ONNX export?

Otherwise I can probably use a Fast or Faster R-CNN for my application, and the roiMaxPooling2D layer is explicitly mentioned in the function documentation as being supported.

Thank you

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Sourabh 2023-5-16

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1960099-export-mask-r-cnn-to-onnx-format#answer_1236869

Hey Karl,

As you have mentioned, the roiAlignLayer is not supported by the exportONNXNetwork function currently. I have attached a few relevant links which might help get you started with your approach using a Fast or Faster R-CNN network:

All the best!