Export Mask R-CNN to ONNX format

7 次查看（过去 30 天）
Karl Mueller
Karl Mueller 2023-5-9
评论： 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

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

采纳的回答

Sourabh
Sourabh 2023-5-16
  1 个评论
Karl Mueller
Karl Mueller 2023-5-18
Thank you,
I have also found that one of the layers in a faster R-CNN is not supported by the ONNX export.
I have designed a fast R-CNN and that should be ok according to the documentation.

请先登录，再进行评论。

更多回答（0 个）

类别

在 Help Center 和 File Exchange 中查找有关 Object Detection 的更多信息

标签

产品


版本

R2022a

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by