运行环境 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

运行环境 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