VS 2019 配置tensorRT生成engine

运行环境
	win11
	cuda 11.1
	tensorRT-8.2.5.1

下载yolov5-6.0版本源代码
下载对应预训练模型预训练模型
下载模型转换文件转换文件

1、转换模型将转换文件中的 gen_wts.py 复制到yolov5的工程文件中,并运行,生成对应模型的.wts文件,用于后续模型序列化。
在这里插入图片描述
2、修改Cmakelist文件,分别添加opencv、tensorrt、以及dirent路径
在这里插入图片描述

3、打开cmake&#

猜你喜欢

转载自blog.csdn.net/qq_44798484/article/details/125422328