Installation

Requirements

• Python >= 3.7
• Numpy
• PyTorch >= 1.9 (Acceleration for 3D depth-wise convolution)
• fvcore: pip install 'git+https://github.com/facebookresearch/fvcore'
• torchvision that matches the PyTorch installation. You can install them together at pytorch.org to make sure of this.
• simplejson: pip install simplejson
• GCC >= 4.9
• PyAV: conda install av -c conda-forge
• ffmpeg (4.0 is prefereed, will be installed along with PyAV)
• PyYaml: (will be installed along with fvcore)
• tqdm: (will be installed along with fvcore)
• iopath: pip install -U iopath or conda install -c iopath iopath
• psutil: pip install psutil
• OpenCV: pip install opencv-python
• torchvision: pip install torchvision or conda install torchvision -c pytorch
• tensorboard: pip install tensorboard
• moviepy: (optional, for visualizing video on tensorboard) conda install -c conda-forge moviepy or pip install moviepy
• PyTorchVideo: pip install pytorchvideo
• Decord: pip install decord

Build UniFormerV2

After having the above dependencies, run:

python setup.py build develop