readme.md update, demo flexible save path (#83)

This commit is contained in:
Iurii Makarov
2024-05-11 15:34:09 +01:00
committed by GitHub
parent 0f9d32869a
commit e29e938311
3 changed files with 10 additions and 4 deletions

View File

@@ -119,7 +119,7 @@ We strongly recommend installing both PyTorch and TorchVision with CUDA support,
git clone https://github.com/facebookresearch/co-tracker
cd co-tracker
pip install -e .
pip install matplotlib flow_vis tqdm tensorboard
pip install matplotlib flow_vis tqdm tensorboard imageio[ffmpeg]
```
You can manually download the CoTracker2 checkpoint from the links below and place it in the `checkpoints` folder as follows:
@@ -132,6 +132,11 @@ cd ..
```
For old checkpoints, see [this section](#previous-version).
After installation, this is how you could run the model on `./assets/apple.mp4` (results will be saved to `./saved_videos/apple.mp4`):
```bash
python demo.py --checkpoint checkpoints/cotracker2.pth
```
## Evaluation
To reproduce the results presented in the paper, download the following datasets: