mmselfSup训练自己的数据集

mmselfsup训练自己的数据集

       最近在做自监督学习的东西，使用无标签数据做预训练模型，做个分享吧，写的不好，请见谅。

        mmselfsup地址：https://github.com/open-mmlab/mmselfsup

       相关文档：Welcome to MMSelfSup’s documentation! — MMSelfSup 0.10.1 documentation

一、环境搭建

1.创建虚拟环境

conda create --name openmmlab python=3.8 -y

激活虚拟环境：

conda activate openmmlab

2.安装pytorch、torchvision

根据自己的配置安装相应版本

pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 -f https://download.pytorch.org/whl/torch_stable.html

或手动下载，地址：https://download.pytorch.org/whl/torch_stable.html

3.下载I MMEngine 和 MMCV

pip install -U openmim
mim install mmengine
mim install 'mmcv>=2.0.0rc1'

4.下载mmselfSup并编译

git clone https://github.com/open-mmlab/mmselfsup.git 
cd mmselfsup
git checkout 1.x
pip install -v -e .

二、训练自监督模型（SimCLR&