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,是有警告的,但不影响结果输出