Improve weights conversion to ONNX

bioimage-io / core-bioimage-io-python

Python libraries for loading, running and packaging bioimage.io models

https://bioimage-io.github.io/core-bioimage-io-python/

MIT License

28 stars 21 forks source link

Improve weights conversion to ONNX #362

Open FynnBe opened 10 months ago

FynnBe commented 10 months ago

Our ONNX converters could be improved.

To convert from pytorch ONNX dynamo looks promising.

oeway commented 9 months ago

I would also like to add that if you convert to ONNX, please set the input shape, on the x, y dimension to -1, currently it causing issue for the BioEngine to support arbitrary input size.