/pytorch-yolov7-deepsort

an implentation of yolov7-deepsort based on pytorch

Primary LanguagePythonMIT LicenseMIT

Deep Sort with Yolov7 in Pytorch

Video

  • Version 1.1

Add tracks in img.

  • Version 1.0 The source code of deep sort is from Deep Sort and the source code of YoloV7-pytorch is from Yolov7-pytorch. You could download corresponding pre-trained weights from the original project pages. When using this code, you may need to adjust the file address of pre-training weight and coco_classes file

Environment

torch>=1.2

Reference

https://github.com/bubbliiiing/yolov7-pytorch

https://github.com/ZQPei/deep_sort_pytorch