Open hgarrereyn opened 4 years ago
I would also love to see 3D ops supported by ONNX.js
For anyone stumbling on this issue, the current workaround I'm using is converting from PyTorch to ONNX to Tensorflow Frozen model to TensorFlow web. Then I can use tensorflow.js to run inference in a browser. It's not elegant but it works.
Looking forward for 3D models support.
I'm trying to export a pytorch model that uses Conv3d layers for in-browser inference with onnx.js. Here is an example model:
Exported via:
In the browser:
This fails with:
If I change
Conv3d
toConv2d
and change the tensor dimensions to[1,2,8,8]
everything works fine. Additionally, the conv3d onnx model works fine in the python onnx runtime so I think this is specifically an onnx.js issue. This may also be related to #161 (unfortunately here there is no workaround by moving up one dimension).