onnx / models

A collection of pre-trained, state-of-the-art models in the ONNX format

http://onnx.ai/models/

Apache License 2.0

7.68k stars 1.37k forks source link

Mask-RCNN weights (h5) to onnx model #454

Open pra-dan opened 3 years ago

pra-dan commented 3 years ago

Ask a Question

Question

I am trying to convert my Mask-RCNN custom models trained on ResNet50 and ResNet101 to onnx (to get an optimised inference using TensorRT). I cannot find any guid to do so. Is there any general/suggested method of doing so ?

Is this issue related to a specific model?
Model name (e.g. mnist): Mask-RCNN

wenbingl commented 3 years ago

please check onnx converter for help

https://lfaifoundation.slack.com/archives/C0171FSKZBN