How to use tensorRT in Yolov5?

I want to optimize my trained model in Yolov5 using tensorRT. But am unable to find a proper way of doing so.Can any body tell show me how to do so?

Upvotes: 0

Answers (3)

Reputation: 3436

Benchmark is used for exporting and evaluating ALL export frameworks. Even the ones that has nothing to do with TenosrRT. Use:

python export.py --weights yolov5s.pt --include engine

for exporting your Yolov5 model to TensorRT

Upvotes: 0

Reputation: 1

would you try:

python benchmarks.py --weights yolov5s.pt --imgsz 640 --device 0

Upvotes: 0

Reputation: 2385

You can refer to this repository for Yolo-V5. It has a section dedicated to tensorrt deployment.

You can also learn about tensorrt inference using C++ and Python

Upvotes: 1