search

Project-MONAI / model-zoo

MONAI Model Zoo that hosts models in the MONAI Bundle format.
Apache License 2.0
191 stars 67 forks source link

ONNX model connector #95

Open diazandr3s opened 2 years ago

diazandr3s commented 2 years ago

Are we considering creating a ONNX connector for MONAI Bundle? This potentially boost MONAI and MONAI Label usability.

Nic-Ma commented 2 years ago

Hi @diazandr3s ,

Thanks for raising it. Actually, ONNX and TorchScript are optional models in a bundle, you can check: https://docs.monai.io/en/latest/mb_specification.html#directory-structure

Thanks.

diazandr3s commented 2 years ago

Superb! Thanks, @Nic-Ma I missed this part Is there any tutorial that we can use as a reference? A MONAI Label user wants to export an ONNX model that also includes functions that are not in MONAI Maybe this should be part of a discussion rather than an issue? :)

Nic-Ma commented 2 years ago

Hi @diazandr3s ,

@binliunls is developing some TorchScript -> ONNX script, will share it soon. And @ericspod may also have some examples.

Thanks.

ericspod commented 2 years ago

I had only briefly experimented with ONNX in the past when discussing Torchscript behaviour: https://gist.github.com/ericspod/65809935adf783c17bdcf19c5568f0fe

diazandr3s commented 2 years ago

Thanks, @ericspod