Exporting PointRend to ONNX

[rvdinter]

Hi, I am trying to export my PointRend model to be used in ML.NET. To my understanding, this can only be done through exporting to ONNX format. Now, I see on this documentation website, that this is not supported. Is there any way I can still export the model to ONNX format, or use the tracing model in ML.NET?

[thiagocrepaldi]

I am working on adding support to ONNX export (with and without caffe2) #4120 and #4153

Hopefully it gets accepted soon

For the PointRend in specific, we will need to add grid_sampler operator to PyTorch, which will be done after these 2 PRs

[thiagocrepaldi]

#4413 is a unit test for PointRend. ONNX Export is already supported by main branch