yolov5导出注意事项
配置
需要提供配置文件和权重文件,不然导出模型不能正常推理。
默认提供检测头。
ModuleNotFoundError: No module named ‘tensorrt’安装TensorRT-python发现报错
由于ModuleNotFoundError: No module named ‘tensorrt’安装TensorRT-python发现报错_modulenotfounderror: no module named 'tensorrt-CSDN博客
在TensorRT的Python文件夹中找到合适的whl,使用pip isntall xxx.whl
pt转engine是先转ONNX,然后再到engine
属性配置
VC++包含目录
VC++目录库目录
如果继承的值中有不需要的,可以将需要的值复制到可编辑的内容中,不勾选从父级或默认项目设置继承 (暂定方法)
连接器--->输入--->附加依赖 ,
opencv_world490d.lib
nvinfer.lib
nvinfer_plugin.lib
nvonnxparser.lib
nvparsers.lib
cublas.lib
cublasLt.lib
cuda.lib
cudadevrt.lib
cudart.lib
cudart_static.lib
cudnn.lib
cudnn64_8.lib
cudnn_adv_infer.lib
cudnn_adv_infer64_8.lib
cudnn_adv_train.lib
cudnn_adv_train64_8.lib
cudnn_cnn_infer.lib
cudnn_cnn_infer64_8.lib
cudnn_cnn_train.lib
cudnn_cnn_train64_8.lib
cudnn_ops_infer.lib
cudnn_ops_infer64_8.lib
cudnn_ops_train.lib
cudnn_ops_train64_8.lib
cufft.lib
cufftw.lib
curand.lib
cusolver.lib
cusolverMg.lib
cusparse.lib
nppc.lib
nppial.lib
nppicc.lib
nppidei.lib
nppif.lib
nppig.lib
nppim.lib
nppist.lib
nppisu.lib
nppitc.lib
npps.lib
nvblas.lib
nvjpeg.lib
nvml.lib
nvrtc.lib
OpenCL.lib
找不到opencv_world490d.dll
在win11系统环境path添加
X:\opencv4.9.0\opencv\build\x64\vc16\bin
重启vs2019.
注意opencv4.9.0没有进行cuda编译,也没有opencv-contrib.
推导代码
参考:TensorRT】TensorRT 部署Yolov5模型(C++)_tensorrt部署yolov5-CSDN博客
【TensorRT】NVIDIA TensorRT 安装 (Windows C++)-CSDN博客
使用上述文章中的代码ONNX转engine速度较慢, engine_file_path需要给出具体名称,如:yolov5_my.engine
运行代码中出现的问题
BUG:Could not locate zlibwapi.dll. Please make sure it is in your library path
下载zlibwapi,将dll_x64中的zlibwapi.dll放入使用的cuda的bin中,将zlibwapi.lib放入lib中,具体参考下面文章:
BUG:Could not locate zlibwapi.dll. Please make sure it is in your library path!-CSDN博客
找不到nvinfer.dll和nvparsers.dll
将TensorRT的的lib加入环境变量中。重启软件。
运行tensorrt显示找不到nvinfer.dll和nvparsers.dll-CSDN博客