Write custom non maximum suppression operator for TorchScript

Hi there,

I’m trying to write a custom nms operator for TorchScript. The original code is here: https://github.com/lucastabelini/LaneATT/tree/main/lib/nms.

My code for building nms operator is here: https://github.com/hthoai/LaneATT/tree/torchscript/lib/nms_torchscript

I caught this error:

Is there anyone know how to fix it?!

It seems that Python.h is missing on your system, so you might want to install e.g. libpython-dev.