Github仓库地址
https://github.com/Monday-Leo/Yolov5_Tensorrt_Win10
项目简介
- 基于Tensorrt加速Yolov5 7.0
- 支持Windows10,Windows11
- 支持C++/Python
电脑环境
- Tensorrt 8.5.2.2
- Cuda 11.6 Cudnn 8.9.2
- Opencv 3.4.6
- Cmake 3.17.1
- VS 2019
- GTX1660Ti
- yolov5-7.0(coco128数据集)
-
生成WTS模型
- 将仓库中的gen_wts.py拷贝至yolov5 7.0的目录下
打开gen_wts.py项目,在项目中修改第11,12行,添加yolov5s.pt权重路径,12行为输出路径改成yolov5-7.0路径下(-w参数为输入pt模型路径,-o参数为输出wts模型的路径。)
根据此图修改后运行,运行完毕后,yolov5目录下会生成wts模型,这个模型用于之后转换为tensorrt专用的序列化模型。