Installment guidance

April 25, 2022 ยท View on GitHub

Requirment

please refer to the requirements.txt file.
recommand to use conda virtual env.
note that some requirements are not so strict.

pip install -r requirements.txt 

Enviroment

pytorch 1.7.0
mmcv-full 1.3.16
cuda 11.4
apex(recommand)
cv2

install mmcv

pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu110/torch1.7.0/index.html  

See here for different versions of MMCV compatible to different PyTorch and CUDA versions.

We apply the backbone of Video-Swin Transformer as the feature extractor. The repo mmaction has provied flexible api to use it. Please refer to it.

Clone the MMAction2 repository.

git clone https://github.com/open-mmlab/mmaction2.git
cd mmaction2

Install build requirements and then install MMAction2.

pip install -r requirements/build.txt
pip install -v -e .  # or "python setup.py develop"

other dependency

pip install tqdm tensorboardX timm einops