参考:https://github.com/visionml/pytracking/blob/master/INSTALL.md
下载
git clone https://github.com/visionml/pytracking/
git submodule update --init
安装(本文尽量使用pip安装)
-
1、创建conda环境
conda create --name pytracking python=3.7
conda activate pytracking
cd pytracking -
2、安装 PyTorch,
pip install torch1.4.0 torchvision0.5.0 -
3、安装很多包和依赖
pip install matplotlib
pip install pandas
pip install tqdm
pip install opencv-python visdom tb-nightly scikit-image tikzplotlib gdown
pip install cython
pip install pycocotools
pip install lvis
sudo apt-get install ninja-build
pip install spatial-correlation-sampler
sudo apt-get install libturbojpeg
pip install jpeg4py -
4、配置环境
python -c "from pytracking.evaluation.environment import create_default_local_file; create_default_local_file()"
python -c "from ltr.admin.environment import create_default_local_file; create_default_local_file()"
mkdir pytracking/networks
- 5、下载与训练模型
- gdown https://drive.google.com/uc?id=1qgachgqks2UGjKx-GdO1qylBDdB1f9KN -O pytracking/networks/dimp50.pth
或者
https://drive.google.com/drive/folders/1WVhJqvdu-_JG1U-V0IqfxTUa1SBPnL0O
手动去这里下载,然后放到pytracking/pytracking/networks下
验证安装成功
cd pytracking
python run_video.py dimp dimp50 /home/zjj/Videos/Download_Video/dog_walk.mp4