Exporting pytorch-ssd to onnx with NonMaxSuppression layer

[orchinia]

Hi,

I'm trying to export pytorch ssd models to onnx. It works fine for me when only exporting SSD modules before nms. However, I stuck at how to modify my pytorch module for exporting to onnx model with NonMaxSuppression layer in it. I implement the nms by myself but it obviously can't be recognize as NonMaxSuppression by onnx.

Is there any suggestions?

Thank you!

[kolingv]

check it here, for the last response. https://github.com/onnx/onnx/issues/2216 it requires custom operators, tutorials is here https://github.com/onnx/tutorials/tree/master/PyTorchCustomOperator I'm currently working on it, but haven't run it sucessfully.

[Edwardmark]

@kolingv @orchinia Have you run nms successfully ? Please help me out, thanks.