ONNX export

[yBeOne]

RuntimeError: Only tuples, lists and Variables are supported as JIT inputs/outputs. Dictionaries and strings are also accepted, but their usage is not recommended. Here, received an input of unsupported type: numpy.ndarray.
Have you export VoxelNeXt to ONNX?

[yukang2017]

Hi,

I think this might be hard. Because the spconv backbone receives sparse tensors, which is not supported yet by ONNX.

I have converted the head of VoxelNeXt to ONNX. That is easy. Please refer to this one.

Regards,
Yukang Chen

[yBeOne]

Hi,
There is another question that how to count the distribution of query voxels like Tabel 7.

[HuangVictorAuto]

Hi,

I think this might be hard. Because the spconv backbone receives sparse tensors, which is not supported yet by ONNX.

I have converted the head of VoxelNeXt to ONNX. That is easy. Please refer to this one.

Regards, Yukang Chen

Hi, Dear Yukang

can you share where is the link or postion for "this one"?

Thanks!

[ArseniuML]

I think this might be hard. Because the spconv backbone receives sparse tensors, which is not supported yet by ONNX.

Could it be done via this plugin?

https://github.com/jingyue202205/SE-SSD-AI-TRT/blob/master/sparseConv3dlayer.h

[Gaussiandra]

Hi,

I think this might be hard. Because the spconv backbone receives sparse tensors, which is not supported yet by ONNX.

I have converted the head of VoxelNeXt to ONNX. That is easy. Please refer to this one.

Regards, Yukang Chen

Could you please provide config to test ONNX model?