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