一、背景
SOLO是一种图像实例分割模型,我们主要下载的是github这个网站SOLO
因为该模型是基于mmdetection跟mmcv来的,所以要多注意学习掌握这两个东东
二、流程
1.根据install.md安装环境
Requirements
此次安装版本
Python 3.8
pytorch 1.10.2 py3.8_cuda11.3_cudnn8.2.0_0 pytorch
先找到Installation部分,打开INSTALL.md开始安装。首先一定要注意Requirements中的各种版本问题,这里面处处是坑。此次SOLO的训练环境是在从NGC下载的pytorch21.07镜像所起的容器中建立的。
a.创建conda虚拟环境并激活
conda create -n solo python=3.8 -y #不是3.7,是3.8,当然以后训练的时候不知道是几了
conda activate solo
这里注意,python版本得是3.8了,要求中的3.7跟其他库的对应版本起冲突
b.按照官方说明安装 PyTorch和torchvision
如果直接按照它给的来,应该是直接装上了最新版本,也是可以的
可以根据official instructions挑选所需要的命令
cond