Hi everybody,
I was able to run yolov8n.dcl based on the code at https://github.com/quic/sample-apps-for-robotics-platforms/tree/master/R...
However, after completing the inference, I have to run the nms function using the CPU, which takes a lot of time. Is there a way to run this nms function using GPU?
I translated the original nms function to cpp code with the torch-cpu library
yolov8's original nms function at https://github.com/ultralytics/ultralytics/blob/main/ultralytics/utils/o...
Thanks,
Tan
I think you should modify your code to use the GPU version of the NMS function. You can utilize the torchvision.ops.nms function, which provides an optimized GPU implementation of NMS. Import the necessary packages: