Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
ShirAmir authored Dec 18, 2021
1 parent 5db962e commit da14c58
Showing 1 changed file with 10 additions and 0 deletions.
10 changes: 10 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,16 @@ We recommend setting the running environment via Anaconda by running the followi
$ conda env create -f env/environment.yml
$ conda activate dino-vit-feats-env
```
Otherwise, run the following commands in your conda environment:
```
$ conda install pytorch torchvision torchaudio cudatoolkit=11 -c pytorch
$ conda install tqdm
$ conda install -c conda-forge faiss
$ conda install -c conda-forge timm
$ conda install matplotlib
$ pip install opencv-python
$ pip install git+https://github.com/lucasb-eyer/pydensecrf.git
```

## ViT Extractor
We provide a wrapper class for a ViT model to extract dense visual descriptors in `extractor.py`.
Expand Down

0 comments on commit da14c58

Please sign in to comment.