1、博客
【SOLO】环境配置(mmdetection=1.0.0+mmcv = 0.2.15)_摇曳的树的博客-CSDN博客
2、实验室主机
3、CRT主机
按照博客【SOLO】环境配置(mmdetection=1.0.0+mmcv = 0.2.15)_摇曳的树的博客-CSDN博客
配置到两次编译之后竟然都非常成功
运行之后报了一点小错在functional.py的这个地方:
在2502行的地方加了这个东西让他不要报错 ,然后就成功运行了inference_demo.py
5、环境配置流程
(1)创建虚拟环境
conda create -n solo python==3.7.3
conda activate solo
(2)安装指定版本的pytorch=1.4.1,torchvision=0.5.0
pip install torch==1.4.0 torchvision==0.5.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
(3)clone版本为 V1.0.0 的mmdetection.(必须是1.0.0版本!!!!!
)
git clone -b v1.0.0 https://github.com/open-mmlab/mmdetection.git
cd mmdetection
安装所需module
conda install opencv
pip install cython numpy albumentations==0.3.2 imagecorruptions matplotlib pytest-runner mmcv==0.2.15 numpy Pillow==6.2.2 six terminaltables asynctest codecov flake8 isort pytest pytest-cov pytest-runner xdoctest==0.10.0 yapf kwarray pycocotools -i https://pypi.tuna.tsinghua.edu.cn/simple
(4)安装完成之后对环境进行编译:
python setup.py develop
(5)在当前mmdetection路径下clone SOLO源码:
git clone https://github.com/WXinlong/SOLO.git
cd SOLO
(6)安装所需module并对环境进行编译:
pip install scipy
python setup.py develop
(7)环境测试:
SOLO目录下创建名为 checkpoints的文件夹,并把demo所需model存入此文件夹
Decoupled_SOLO_R50_3x:
https://cloudstor.aarnet.edu.au/plus/s/dXz11J672ax0Z1Q/download
运行测试指令:
cd demo
python inference_demo.py
到这里的时候因为版本问题报错了,torchvision0.4以上有这个问题,跟代码不匹配,在上面CRT主机部分2502添加一行align_corners = True来修复