SOLOV2环境配置(mmdetection版

1. 安装cuda （可参考https://blog.csdn.net/weixin_43848614/article/details/117221384）

查看显卡

下载两个安装包：

cuda

cudnn

根据我的显卡下载的：

2

 #创建虚拟环境
 
   conda create -n solo python=3.7 -y
   conda activate solo

 #安装pytorch=1.7.0 

   conda install pytorch==1.7.0 torchvision cudatoolkit=10.2 -c pytorch

 #安装mmcv(官方文档是0.2.16）

   pip install mmcv==0.2.16 

 #安装mmdection，安装命令参考官网https://github.com/open-mmlab/mmdetection
 
  (如果使用git clone命令,先安装Git,官网: https://git-scm.com/,按照步骤点击下一步安装就可以)
 
   git clone https://github.com/open-mmlab/mmdetection.git 
   (有时候下载失败,多试几次可以成功）
   cd mmdetection
   pip install -r requirements/build.txt
   pip install -v -e . (注意e的后面有空格和.) # or "python setup.py develop"

 #SOLO 参考https://github.com/WXinlong/SOLO/blob/master/docs/INSTALL.md
   
   git clone https://github.com/WXinlong/SOLO.git
   cd SOLO
   pip install -r requirements/build.txt 
   pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI
   pip install -v -e .  # or "python setup.py develop"



  

 环境搭建成功了！

 3.测试一下是否搭建成功：

1.下载权重文件，推荐SOLOv2_X101_DCN_3x

2.修改inference_demo.py文件并运行

# 在打开的 inference_demo.py文件中修改这两行

config_file = '../configs/solov2/solov2_x101_dcn_fpn_8gpu_3x.py'
checkpoint_file = '../SOLOv2_X101_DCN_3x.pth'

#可以把你想要测试的图片放到demo文件夹中
img = '自己想要测试图片的名字'   

#demo文件夹里得到了demo_out.jpg是输出的图片结果

在这里我运行demo，是有警告的，但不影响结果输出