search

Eruvae / yolact_ros

ROS wrapper for yolact instance segmentation
52 stars 45 forks source link

Running yolact_ros in TensorRT #28

Open bibekyess opened 1 year ago

bibekyess commented 1 year ago

Hello! Thanks for this awesome work.

Is it possible to run yolact_ros in TensorRT? I am running yolact_ros in JetsonNX and it uses almost all of GPU during inference and I am doing real-time object detection so I want to optimize it using tensorRT. Have anyone tried running yolact_ros in tensorRT? Is it possible? Can you give me some hint or useful documentation?

Thank you for your help!

Eruvae commented 1 year ago

YolactEdge supports TensorRT, so you could try setting that up and then using the yolact_edge_ros script that I added for issue #16. I have not used it myself though, so no guarantee that it works out of the box.