多目标跟踪yolov5+deep sort转ncnn调用

deep sort c++开源代码情况:
https://blog.csdn.net/qq_35975447/article/details/119780276

1.C++调用代码
选择shaoshengsong的C++调用代码
https://github.com/shaoshengsong/DeepSORT

这里需要去掉tensorflow相关的代码,下载并添加Eigen代码路径,其他需要的库:opencv(mingw的,可从github找到,需要带ffmpeg的,否则播放不了视频,我这里用的OpenCV-MinGW-Build-OpenCV-3.4.6),ncnn(编译与使用可以看我其他的博客)

修改之后的yolov5+deep sort调用C代码放到了github(其中用到少量Qt库,可根据需要去掉Qt相关的部分):
https://github.com/Whu-wxy/yolov5_deepsort_ncnn_qt/

2.模型转换
yolov5模型训练与转换(yolov5在2021年10月多更新的第六版去掉了输入的focus层,C++调用很方便了):
https://github.com/ultralytics/yolov5

deep_sort pytorch模型转为ncnn:
用下面这里的模型文件转onnx的python代码
https://github.com/RichardoMrMu/deepsort-tensorrt
转之前需要把
https://github.com/ZQPei/deep_sort_pytorch
里面和fastreid部分先删掉,就不用配置fastreid了,因为用不到

转的时候还需要注意模型的路径和configs/deep_sort.yaml文件里写的路径一致

3.onnx转ncnn
https://convertmodel.com/#outputFormat=ncnn

4.模型文件
转换后的ncnn模型文件与测试视频放到了百度网盘, 提取码:xsuk。我是把视频,模型文件都放到了pro文件同级的src目录下面。

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值