安装mmsegmentation框架-有手就行!
框架地址:https://github.com/open-mmlab/mmsegmentation
一.安装pytorch cuda torchvision
1.通过命令行查看驱动版本及可支持的最高CUDA版本
Win+R输入cmd打开命令行,输入命令NVIDIA-smi 查看
2.pytorch 官网 https://pytorch.org/ 复制对应的命令 如下图所示:
pip3 install install torch==1.10.1+cu113 torchvision==0.11.2+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.htm
我是安装了 PyTorch(1.10.1,针对 CUDA 11.3)和 torchvision(0.11.2,针对 CUDA 11.3)。-f 参数指定了从 PyTorch 官方的特定镜像源下载安装文件。
二.安装openmim mmengine mmcv
pip install -U openmim
mim install mmengine
mmcv与cuda、pytorch版本兼容要求,见mmcv官方文档:
官网:https://mmcv.readthedocs.io/zh-cn/latest/get_started/installation.html#pip
mim install mmcv==2.0.0
三.安装opencv-python、Pillow、matplotlib、seaborn、tqdm、pytorch-lightning 和 mmdet
pip install opencv-python pillow matplotlib seaborn tqdm pytorch-lightning mmdet>=3.1.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
四.安装 setuptools 和 prettytable
pip install setuptools==59.5 prettytable
setuptools 通常用于Python包的安装和管理,而 prettytable 是一个用于在终端中打印漂亮表格的库。
五.安装 安装 MMSegmentation
git clone https://github.com/open-mmlab/mmsegmentation.git -b v1.2.2#替换你需要的版本 或者自己手动下载
cd mmsegmentation
pip install -v -e .
跑mask2former试一下
python tools\train.py configs/mask2former/mask2former_swin-b-in1k-384x384-pre_8xb2-160k_ade20k-640x640.py
安装ftfy
ftfy是一个Python包,它的主要功能是自动检测并修复包含错误编码、乱码或其他Unicode问题的文本。这个工具包可以帮助处理那些由于编码问题导致的文本显示异常,通过修复这些问题,使得文本能够正确显示。
conda install ftfy
看了下预训练权重下载到这个地址了:C:\Users\user.cache\torch\hub\checkpoints
其实可以手动下载 放到pretrain文件夹里
成功!!!!!