requirements.txt

--extra-index-url https://download.pytorch.org/whl/cu118
torch==2.1.0
torchvision
scipy
opencv-python
timm
transformers
einops
scikit-video
git+https://github.com/facebookresearch/detectron2.git