SpaTracker&CoTracker 环境配置

spa

下载地址

https://github.com/isl-org/MiDaS #Accuracy

https://github.com/isl-org/ZoeDepth/releases

https://drive.google.com/drive/folders/1UtzUJLPhJdUg2XvemXXz1oe6KUQKVjsZ

配置命令

conda create --name spa_ python=3.10 cudatoolkit=11.8

conda activate spa_

pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118

pip install -r requirements.txt
# cupy会报错所以先注释掉

pip install C:\Users\HP\Downloads\cupy_cuda12x-13.2.0-cp310-cp310-win_amd64.whl

运行demo

python demo.py --model spatracker --downsample 1 --vid_name butterfly --len_track 1 --fps_vis 15  --fps 1 --grid_size 40

co

下载地址

https://github.com/facebookresearch/co-tracker

配置命令

conda activate spa_

# 可直接使用spa的环境


# 建一个 checkpoints 文件夹，从下面网址手动下载放进去：
# https://huggingface.co/facebook/cotracker/resolve/main/cotracker2.pth

运行demo

python demo.py --checkpoint checkpoints/cotracker2.pth