SOLOV2环境配置：Ubuntu18.04

文章目录[隐藏]

环境配置
运行demo 测试自己的图片
出错

SOLOV2环境配置：Ubuntu18.04

环境配置

运行SOLOV2 demo 测试自己的图片

配置了几次的solov2 决定把配置过程记录一下
主要参考了以下几篇文章：
https://blog.csdn.net/weixin_40249907/article/details/109010880
https://blog.csdn.net/weixin_40249907/article/details/109010880
https://blog.csdn.net/weixin_40249907/article/details/109010880

基础配置：
系统：Ubuntu18.04
显卡：2080TI
cuda 10.1
还有说别的 opencv-python 4.1.1.26 gcc 7.5.0 什么的我没有管，也成功了

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

#安装 pytorch==1.4.0 torchvision==0.5.0  试了别的1.7.0啥的都没成功
conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1 -c pytorch

#安装mmcv，安装参考官方https://github.com/open-mmlab/mmcv/tree/v0.2.16
pip install mmcv-full

#安装mmdection，安装命令参考官方https://github.com/open-mmlab/mmdetection
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -r requirements/build.txt
pip install -v -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     #solo文件中在requirements文件夹中将torch版本相关注释掉，防止重新安装，上面已经安装了1.4.0了
pip install "git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI"
pip install -v -e .  # or "python setup.py develop"

OK，到此已经环境搭建好了，下面开始测试自己的图片

运行demo 测试自己的图片

1、下载权重文件
下载的是另一个博主提供的精度最高的SOLOv2_X101_DCN_3x
2、修改inference_demo.py文件

cd SOLO/demo
gedit inference_demo.py

#在打开的 inference_demo.py文件中修改这两行
config_file = '../configs/solov2/solov2_x101_dcn_fpn_8gpu_3x.py'
checkpoint_file = '../SOLOv2_X101_DCN_3x.pth'
img = '自己想要测试图片的名字'   #把你想要测试的图片放到demo文件夹中

运行

cd demo   #进入到demo文件夹中
python inference_demo.py

这样就可以测试自己的图片啦！！！成功运行demo

出错

1、ModuleNotFoundError: No module named 'mmdet
第一次没有单独安装mmcv和mmdection 找不到这个，后来先安装好mmcv和mmdection这两个就解决了
2、torch.xxx报错
cuda10.1+pytorch1.4.0 torchvision0.5.0 cudatoolkit=10.1 才可以
pytorch==1.7.0 就会报错卸载重新装的
3、运行的时候有个warning 什么nn.啥的，没管他，反正也能跑

有什么问题欢迎在下面评论！看到了会及时回复的

版权声明：本文为CSDN博主「DuanFa95613」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/weixin_44951291/article/details/111238585