工具介绍
使用 YOLOv4、DeepSort 和 TensorFlow 实现物体跟踪。
yolov4-deepsort 使用 YOLOv4、DeepSort 和 TensorFlow 实现物体跟踪。YOLOv4 是一项先进的算法,它使用深度卷积神经网络来进行物体检测。我们可以将 YOLOv4 的输出传递给 Deep SORT(简单的在线和实时跟踪,使用深度关联度量),以创建高精度的物体跟踪器。
开始使用
要开始使用,请通过 Anaconda 或 Pip 安装相应的依赖项。我建议使用 Anaconda 途径,因为它会为您配置 CUDA 工具包版本。
下载官方 YOLOv4 预训练权重 我们的目标跟踪器使用 YOLOv4 进行目标检测,然后由 deep sort 进行跟踪。存在一个官方预训练的 YOLOv4 目标检测器模型,可以检测 80 个类别。为了方便演示,我们将使用该模型的预训练权重。下载预训练的 yolov4.weights 文件: https://drive.google.com/open?id=1cewMfusmPjYWbrnuJRuKhPMwRe_b9PaT