边缘设备jetson或x64(ubuntu22.04)部署yolov5-deepsort目标追踪踩坑记录

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

  • 27
    点赞
  • 23
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值