Yolov5 +使用PyTorch进行深度排序
介绍
该存储库包含PyTorch YOLOv5的简化版( )。 它过滤掉不是人的所有检测。 然后,将对人员的检测传递给跟踪人员的深度排序算法( )。 它仅跟踪人员这一事实背后的原因是,深度关联度量仅在人员数据集上进行训练。
描述
该实现基于两篇文章:
使用深度关联指标进行简单的在线和实时跟踪
YOLOv4:物体检测的最佳速度和准确性
要求
安装了所有requirements.txt依赖关系的Python 3.8或更高版本,包括torch> = 1.7。 要安装运行:
pip install -U -r requirements.txt
所有依赖项都包含在关联的docker映像中。 Docker要求是:
nvidia-docker
Nvidia驱动程序版本> = 440.44
在运行跟踪器之前
递归克隆存储库:
git clo
1