core-bioimage-io-python icon indicating copy to clipboard operation
core-bioimage-io-python copied to clipboard

Published 20 hours ago • verified publisher icon bioimage-io

Improve weights conversion to ONNX

Open FynnBe opened this issue 1 year ago • 1 comments

Our ONNX converters could be improved.

To convert from pytorch ONNX dynamo looks promising.

FynnBe avatar Nov 15 '23 23:11 FynnBe

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.

oeway avatar Dec 05 '23 16:12 oeway