Write custom non maximum suppression operator for TorchScript

jit
1

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:

image
image1393×371 45.6 KB

Is there anyone know how to fix it?!

2

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