yolov5

YOLOv5 PyTorch=>ONNX=>TensorRT

1.Reference

• yolov5: https://github.com/ultralytics/yolov5
• get yolov5 weights from here: yolov5s, yolov5m, yolov5l, yolov5x.

2.Export ONNX Model

git clone https://github.com/linghu8812/yolov5.git
python3 export.py ---weights weights/yolov5s.pt --batch-size 10 --imgsz 640 --include onnx --simplify

3.Build yolov5_trt Project

mkdir build && cd build
cmake ..
make -j

4.Run yolov5_trt

• inference with yolov5s

./yolov5_trt ../config.yaml ../samples

• inference with yolov5s6

./yolov5_trt ../config6.yaml ../samples

For more information, please refer this blog: https://blog.csdn.net/linghu8812/article/details/109322729

5.Results: