Using the tensorrt model in python

Robotics & Edge Computing Jetson & Embedded Systems Jetson Nano
,
11

Hi,

Based on the following code:

model = torch.hub.load(‘ultralytics/yolov5’, ‘custom’, path=‘enson.engine’,force_reload=True)

Do you try to load a TensorRT engine with PyTorch API?
If yes, have you tried it before on other platforms?

In general, we load an engine with TensorRT API directly.
You can find an example in the below link:
https://elinux.org/Jetson/L4T/TRT_Customized_Example#OpenCV_with_PLAN_model

Thanks.