Forums - How to run nms function with gpu to speed up time on RB5 dev kit. After inference using model_quant.dcl file

1 post / 0 new
or Register
How to run nms function with gpu to speed up time on RB5 dev kit. After inference using model_quant.dcl file
nguy3nt4n99
Join Date: 14 Mar 24
Posts: 2
Posted: Sat, 2024-04-27 22:45
Top

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


Thanks,
Tan

  • Up0
  • Down0

Opinions expressed in the content posted here are the personal opinions of the original authors, and do not necessarily reflect those of Qualcomm Incorporated or its subsidiaries (“Qualcomm”). The content is provided for informational purposes only and is not meant to be an endorsement or representation by Qualcomm or any other party. This site may also provide links or references to non-Qualcomm sites and resources. Qualcomm makes no representations, warranties, or other commitments whatsoever about any non-Qualcomm sites or third-party resources that may be referenced, accessible from, or linked to this site.