Hi @tom thank you so much for your reply. I can not use nms from torchvision since I am working on 3d data and not 2d images. However, if you can still direct me to actual source code of nms then maybe I can draw some inspiration from it. My C++ skills are very basic and as such, I do need a lot of help in completing this task.
Thanks again,
Chinmay