Installation

We provide the instructions to install the dependency packages.

Requirements

We test the code in the following environments, other versions may also be compatible:

• CUDA 11.1
• Python 3.7
• Pytorch 1.8.1

Setup

First, clone the repository locally.

https://github.com/OpenGVLab/MUTR.git

Then, install Pytorch 1.8.1 using the conda environment.

conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 -c pytorch

Install the necessary packages and pycocotools.

pip install -r requirements.txt 
pip install 'git+https://github.com/facebookresearch/fvcore' 
pip install -U 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'

Finally, compile CUDA operators.

cd models/ops
python setup.py build install
cd ../..