YoloV5+DeepSort+TensorRT 目标检测、跟踪

文章发布已转个人主页：wangcong.net

更新(Update !):

对DeepSORT中的特征提取部分进行了加速！
P.S.C++测试版本项目地址：https://github.com/cong/yolov5_deepsort_tensorrt_cpp

---

介绍

项目通过 YOLOv5 和 DeepSORT 来实现了目标检测、跟踪算法，其中基于TensorRTX 实现了模型从 PyTorch 到 TensorRT 转换，进一步将代码部署 NVIDIA Jetson Xavier NX 中。

项目在 NVIDIA Jetson Xavier NX 和 X86 平台 都可以正常运行。

项目地址：https://github.com/cong/yolov5_deepsort_tensorrt

环境依赖

1. the X86 architecture:
   • Ubuntu20.04 or 18.04 with CUDA 10.0 and cuDNN 7.6.5
   • TensorRT 7.0.0.1
     - PyTorch 1.7.1_cu11.0, TorchVision 0.8.2+cu110, TorchAudio 0.7.2
   • OpenCV-Python 4.2
   • pycuda 2021.1
2. the NVIDIA embedded system：
   • Ubuntu18.04 with CUDA 10.2 and cuDNN 8.0.0
   • TensorRT 7.1.3.0
     - PyTorch 1.8.0 and TorchVision 0.9.0
   • OpenCV-Python 4.1.1
   • pycuda 2020.1

测试速度

the X86 architecture with RTX 2080Ti :</