[jetson浅试] yolov5+deepsort+Tensorrt C++部署(Xavier AGX)

1.简介:

这学期刚开学的时候搞的,空下来整理一些(以后还是应该养成边搞边写博客的好习惯)

本文主要是对yolov5-deepsort-tensorrt: A c++ implementation of yolov5 and deepsort (gitee.com)中的内容进行复现,熟悉xavier的配置流程,以及对xavier算力有一个相对直观的认识

2.使用平台介绍:

使用的平台是天准的Xavier AGX,出厂包含了opencv和tensorRT,nv自带的工具也可以安装,这部分还是比较方便的。

3.流程介绍:

3.1 clone yolov5

git clone -b v5.0 https://github.com/ultralytics/yolov5.git
cd yolov5
mkdir weights
cd weights
// download https://github.com/ultralytics/yolov5/releases/download/v5.0/yolov5s.pt
wget https://github.com/ultralytics/yolov5/releases/download/v5.0/yolov5s.pt

3.2 clone tensorrtx

git clone https://github.com/wang-xinyu/tensorrtx

3.3 使用tensorrtx将转换模型为wts

cp tensorrtx/gen_wts.py yolov5/
cd yolov5 
python3 gen_wts.py -w ./weights/yolov5s.pt -o ./weights/yolov5s.wts
// a file 'yolov5s.wts' will be generated.

3.4 项目编译

cd tensorrtx/yolov5
// update CLASS_NUM in yololayer.h if your model is trained on custom dataset
mkdir build
cd build
cp {ultralytics}/yolov5/yolov5s.wts {tensorrtx}/yolov5/build
cmake ..
make -j8

3.5 转换权重为 engine文件

// yolov5s
sudo ./yolov5 -s yolov5s.wts yolov5s.engine s
// test your engine file
sudo ./yolov5 -d yolov5s.engine ../samples

记住engine位置或者放入项目中指定位置

3.6 修改engine位置,测试demo

git clone https://github.com/RichardoMrMu/deepsort-tensorrt.git
// 根据github的说明
cp {deepsort-tensorrt}/exportOnnx.py {deep_sort_pytorch}/
python3 exportOnnx.py
mv {deep_sort_pytorch}/deepsort.onnx {deepsort-tensorrt}/resources
cd {deepsort-tensorrt}
mkdir build
cd build
cmake ..
make 
./onnx2engine ../resources/deepsort.onnx ../resources/deepsort.engine
// test
./demo ../resource/deepsort.engine ../resources/track.txt

3.7 运行结果

我测试比作者给出来的结果高不少(毕竟他用的nx,而且我测试的场景中,目标数量相对较少)

 

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值