1、前言
下载测试大佬的目标追踪项目部署踩坑记录,链接:Jetson 系列——基于yolov5和deepsort的多目标头部识别,跟踪,使用tensorrt和c++加速_no module named 'fastreid-CSDN博客
在新的环境中测试下,实验环境x64pc+ubuntu22.04,TensorRT8.5.1.7, cuda11.8;原作者使用的是TensorRT7等较老的环境。
2、报错记录
实验需要git下来deep_sort_pytorch, deepsort_tensorrt, tensorrtx-yolov5-v5.0, yolov5-v5.0, yolov5-deepdort-tensorrt-main五个库,按照步骤需要执行即可正常进行。
很多报错都能够在这位博主的文档中找到:用C++部署yolov5+deepsort+tensorrt实现目标跟踪_基于tensorrt部署yolov5 c++-CSDN博客
主要的报错记录也同样出现在原作者的评论中。
yolov5_trt_create ...
yolov5_trt_create cuda engine...
[08/17/2022-19:51:16] [E] [TRT] 3: Cannot find binding of given name: data
[08/17/2022-19:51:16] [E] [TRT] 3: Cannot find binding of given name: prob
yolov5_trt_create buffer ...
yolov5_trt_create stream ...
yolov5_trt_create done ...
create yolov5-trt , instance = 0x5594955b60
origin img size:1920 1080
yolov5_trt_detect start preprocess img
yolov5_trt_detect start do inference
CUDA error 1 at /home/nvidia/deepsort/yolov5-deepsort-tensorrt-main/yolo/src/yolov5_lib.cpp:30Segmentation fault (core dumped)
主要原因是自己使用trtexec通过yolov5s.onnx生成yolov5s.engine,并没有使用作者提供的tensortx-yolov5仓库按照步骤生成yolov5s.wts生成engine,重新生成了下然后通过最后的编译。就在这里卡了好久。
CmakeLists.txt添加tensorrt8,cuda11.8并没有太大的报错问题。
3、总结
后续再测试下yolov8等其它版本的。感谢大佬们提供的优质链接。
参考链接:https://blog.csdn.net/m0_37605642/article/details/122648601
https://blog.csdn.net/weixin_42264234/article/details/120152117#t2