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.