Files
IDGallagher-MotionVideoSearch/README.md
T
2025-01-03 16:27:38 +08:00

1.1 KiB

Installation

Set Up Environment

Clone the repository.

git clone https://github.com/16lemoing/dot
cd dot
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 torchvision torchaudio --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

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