搭建MOTRv2

MealHunterMealHunter
1 min read

1、创建环境

不要根据文档的说明进行安装,要安装python=3.8版本才行

conda create -n motrv2 python=3.8
conda activate motrv2
pip install torch==1.8.1+cu101 torchvision==0.9.1+cu101 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

安装其他库

pip install -r requirements.txt

2、拉取代码

git@github.com:megvii-research/MOTRv2.git

3、运行步骤

3.1、安装MultiScaleDeformableAttention

首先到models/ops这个目录下运行

./make.sh

0
Subscribe to my newsletter

Read articles from MealHunter directly inside your inbox. Subscribe to the newsletter, and don't miss out.

Written by

MealHunter
MealHunter