51 lines
1.4 KiB
Markdown
51 lines
1.4 KiB
Markdown
## Installation
|
|
|
|
### Set Up Environment
|
|
Clone the repository.
|
|
```
|
|
git clone https://github.com/IDGallagher/MotionVideoSearch
|
|
cd MotionVideoSearch
|
|
```
|
|
|
|
### Install dependencies
|
|
|
|
[Optional] Create a conda environment.
|
|
```
|
|
conda create -n mvs python=3.9
|
|
conda activate mvs
|
|
```
|
|
|
|
Install the [PyTorch and TorchVision](https://pytorch.org/get-started/locally/) versions which are compatible with your CUDA configuration.
|
|
```
|
|
pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu118
|
|
```
|
|
|
|
[Optional] Install a version of xformers that works with your version of PyTorch
|
|
```
|
|
pip install xformers --index-url https://download.pytorch.org/whl/cu118
|
|
```
|
|
|
|
Install ffmpeg 6.x
|
|
```
|
|
conda install -c conda-forge ffmpeg=6.*
|
|
```
|
|
|
|
Install other dependencies.
|
|
```
|
|
pip install tqdm matplotlib einops einshape scipy timm lmdb av mediapy typer imageio imageio-ffmpeg requests opencv-python
|
|
|
|
```
|
|
Install faiss https://github.com/facebookresearch/faiss
|
|
```
|
|
conda install -c pytorch faiss-cpu=1.9.0
|
|
```
|
|
|
|
[Optional] Set up custom modules from [PyTorch3D](https://github.com/facebookresearch/pytorch3d) to increase speed and reduce memory consumption of interpolation operations.
|
|
```
|
|
cd dot/utils/torch3d/ && python setup.py install && cd ../../..
|
|
```
|
|
|
|
[Optional] Download watermark removal repo
|
|
```
|
|
git clone https://github.com/l-comm/WatermarkRemoval.git
|
|
``` |