search

NVIDIA-AI-IOT / trt_pose

Real-time pose estimation accelerated with NVIDIA TensorRT
MIT License
972 stars 287 forks source link

Running on DLA #109

Open oscarthorn opened 3 years ago

oscarthorn commented 3 years ago

Hi!

Is it possible to run trt_pose on one of the DLA cores on a Xavier NX? Specifically the densenet version?

jaybdub commented 3 years ago

Hi @oscarthorn,

Thanks for reaching out!

We may need to update torch2trt to support this directly. That said, I believe this model may be exported to ONNX and use other TensorRT tools for this purpose.

Please let me know if this helps or you have any questions.

Best, John

mohammadhaqqi commented 2 years ago

Did you find the solution