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 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 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