淘先锋技术网

首页 1 2 3 4 5 6 7

参照YOLO v7 + 各种tracker实现多目标跟踪配置环境

1、配置要求

  • python=3.7.0 pytorch=1.7.0 torchvision=0.8.0 cudatoolkit=11.0
  • py-motmetrics (pip install motmetrics)
  • cython-bbox (pip install cython_bbox)
  • opencv

2、配置过程

创建py37虚拟环境

conda create -n uavMOT python=3.7.0

进入虚拟环境uavMOT

conda activate uavMOT

参考Ubuntu20.04安装 Pytorch1.7.0 GPU版 torchvision=0.8.1 (已装好CUDA和CUDNN),下载链接在https://download.pytorch.org/whl/cu101/torch/中,对应找到即可。将下方中的cp38改为cp37

pip install torch-1.7.0+cu101-cp37-cp37m-linux_x86_64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple/

下方指令完全参照Ubuntu20.04安装 Pytorch1.7.0 GPU版 torchvision=0.8.1 (已装好CUDA和CUDNN)来执行

pip install torch==1.7.0+cu101 torchvision==0.8.0+cu101 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

torchvision和cuda版本有调整,其检测过程如下:

>>> import torch
>>> print(torch.__version__)
1.7.0+cu101
>>> import torchvision
>>> print(torchvision.__version__)
0.8.1+cu101
>>> print(torch.version.cuda)
10.1
>>> from torch.backends import cudnn
>>> print(cudnn.is_available())
True
>>> print(torch.cuda.is_available())
True

下方指令需要执行很久

pip install motmetrics

执行下方指令时报错ModuleNotFoundError: No module named 'Cython',使用pip install Cython安装cython时报错ERROR: Could not find a version that satisfies the requirement Cython (from versions: none) ERROR: No matching distribution found for Cython ,参考链接得以成功安装cython

pip install cython_bbox