项目地址:https://github.com/dvlab-research/MSAD
一 实验环境
- ubuntu 18.04
- CUDA 10.2
- python 3.6
- pytorch 1.8.0
- torchvision 0.9.0
二 安装步骤
1.先创建虚拟环境:
conda create -n msad python=3.6
conda activate msad
2.安装pytorch
conda install pytorch==1.8.0 torchvision torchaudio cudatoolkit=10.2 -c pytorch
3. MSAD是基于Detectron2 做的,所以要先下载安装Detectron2
(1) 下载并解压detectron2-main.zip: unzip detectron2-main.zip