【目标跟踪】Yolov5_DeepSort_Pytorch中使用不同版本的yolov5

1.环境

ubuntu16.04
cuda10.1
cudnn7
python3.6
 
 
Cython
matplotlib>=3.2.2
numpy>=1.18.5
opencv-python>=4.1.2
Pillow
PyYAML>=5.3
scipy>=1.4.1
tensorboard>=2.2
torch>=1.7.0  (my 1.7.1)
torchvision>=0.8.1 (my 0.8.2)
tqdm>=4.41.0
seaborn>=0.11.0
easydict
thop
pycocotools>=2.0

2.track.py

import sys
sys.path.insert(0, './yolov5_vx') # vx表示release的版本

from yolov5_vx.utils.datasets import LoadImages, LoadStreams
from yolov5_vx.utils.general import check_img_size, non_max_suppression, scale_coords
from yolov5_vx.utils.torch_utils import select_device, time_synchronized