一、MMClassfication
MMClassfication 已经支持大量的模型,也支持了主流的图像分类数据集。
MMClassfication仓库地址:https://github.com/open-mmlab/mmclassification
MMClassfication教程地址:https://mmclassification.readthedocs.io/en/latest/
MMClassfication主要通过构建配置文件进行,简化了直接手写Pytorch代码的过程。
二、使用北京超算平台配置MMClassfication环境
1、官网(北京超级云计算中心)下载客户端 并打开 , 输入账号密码
2、打开SSH
3、进入run目录
cd run/
4、加载Anaconda工具
module availmodule load anaconda/2021.05
5、创建MMCls环境
创建虚拟环境
conda create --name mmclassfication python=3.8
激活虚拟环境
source activate mmclassfication
安装pytorch
pip install torch==1.10.0+cu111 torchvision==0.11.0+cu111 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html
加载cuda
module load cuda/11.1
安装mmcv-full
pip install mmcv-full==1.7.0 -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.10/index.html
安装mmclassfication
mkdir mmclassficationcd mmclassficationgit clone https://github.com/open-mmlab/mmclassification.gitcd mmclassficationmodule load gcc/7.3pip install -v -e .
一、MMClassification 介绍
MMClassification ---- 图像分类工具包