Tensorrt YOLOV3 踩坑记录

Tensorrt YOLOV3 踩坑记录

本篇用于记录自己在测试tensorrt时踩过的一些坑，防止以后重复类似错误。
ubuntu18.04
tensorrt->7.0.0.11
cuda->10.1
cudnn->7.6
源码地址：https://github.com/wang-xinyu/tensorrtx

1.fatal error: NvInfer.h: No such file or directory

这个错误主要是CMakelist.txt中Tensorrt的地址有误导致，将：
tensorrt的路径修改为你的地址即可。

set(TENSORRT_HOME "/usr/local/TensorRT")
# tensorrt
include_directories(${TENSORRT_HOME}/include)
link_directories(${TENSORRT_HOME}/lib)

2.opencv2/dnn/dnn.hpp can not find

这个问题困扰了很久，当时使用的opencv版本为4.0，已经支持DNN库了，但源文件中找不到dnn相关的内容，应该是opencv安装时就出现的问题。最后的办法就是，只能重装opencv…
将版本切换为opencv4.1，具体重装步骤参考博客：在ubuntu下安装opencv4.1.0详细步骤

3.Parameter check failed at: …/builder/Network.cpp

导致这个的原因是与训练模型与网络结构不匹配，重新生成对应的预训练模型即可。