1.首先在官网上下载代码以及数据集。代码下载如下:
git clone +URL网址
git clone https://github.com/facebookresearch/adaptive_teacher.git
2.安装环境
这是作者给的readme。
2.1 首先创建环境,这里我创建的python版本为3.7。
conda create -n AT python=3.7
2.2 安装pytorch、torch vision以及detectron2
这里安装v0.6版本的detectron2代码不能运行,于是去安装v0.5版本的detectron2。
我服务器的cuda是11.6。这里我安装cuda11.0,torch=1.7以及对应版本的torchvison。
conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=11.0 -c pytorch