tensorRT 部署 YOLOV5模型详解
第一步:
下载tensorRT库
https://developer.nvidia.com/nvidia-tensorrt-8x-download
解压后,编译
第二步:cmakeList.txt需要进行修改,·改成自己的路径
第三步:下载dirent-master 将dirent.h文件copy到src文件夹下
第四步:使用的opencv_world460.dll 拷贝进环境变量
第五步 参数进行修改
第六步
6.1、pt->wts
python gen_wts.py -w yolov5s.pt -o yolov5s.wts
6.2、wts -----> engine
yolov5_det.exe -s yolov5l_best.wts yolov5l_best.engine l
第七步 保持yolov5与tensorrt版本一致
第八步、cmake即可,打开生成工程
总结:过程中需要注意两点:a、主要配置自己的参数;b、cmakeLists文件中的路径修改;