安装mmsegmentation默认主分支main

安装时间2024.4.21

mmsegmentation新版本main分支(v1.2.2)

安装过程

python 复制代码
conda create --name openmmlab python=3.8 -y
conda activate openmmlab

// 很关键,可以避免mmcv版本问题
pip install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

pip install -U openmim
mim install mmengine
mim install "mmcv>=2.0.0"

git clone -b main https://github.com/open-mmlab/mmsegmentation.git
cd mmsegmentation
pip install -v -e .
# '-v' 表示详细模式,更多的输出
# '-e' 表示以可编辑模式安装工程,
# 因此对代码所做的任何修改都生效,无需重新安装

验证和测试

python 复制代码
mim download mmsegmentation --config pspnet_r50-d8_4xb2-40k_cityscapes-512x1024 --dest .

python demo/image_demo.py demo/demo.png configs/pspnet/pspnet_r50-d8_4xb2-40k_cityscapes-512x1024.py pspnet_r50-d8_512x1024_40k_cityscapes_20200605_003338-2966598c.pth --device cuda:0 --out-file result.jpg

将在当前文件夹中看到一个新图像 result.jpg,其中所有目标都覆盖了分割 mask

最大的问题在于mmcv,如果需要安装时,或者更改版本时,建议先卸载

复制代码
pip uninstall mmcv
pip uninstall mmcv-full
相关推荐
若兰幽竹21 分钟前
【从零开始编写数据库:基于Python语言实现数据库ToyDB的ACID特性】
数据库·python
xiaocainiao88129 分钟前
Python 实战:构建 Git 自动化助手
git·python·自动化
nightunderblackcat44 分钟前
新手向:使用Python将多种图像格式统一转换为JPG
开发语言·python
Liudef061 小时前
三维点云Transformer局部感受野构建:理论、方法与挑战
人工智能·深度学习·transformer
engchina1 小时前
Python PDF处理库深度对比:PyMuPDF、pypdfium2、pdfplumber、pdfminer的关系与区别
开发语言·python·pdf
Lo-Y-eH1 小时前
Openpyxl:Python操作Excel的利器
python·excel
DAWN_T171 小时前
Transforms
pytorch·python·机器学习·jupyter·pycharm
一百天成为python专家2 小时前
python库之jieba 库
开发语言·人工智能·python·深度学习·机器学习·pycharm·python3.11
27669582922 小时前
tiktok 弹幕 逆向分析
java·python·tiktok·tiktok弹幕·tiktok弹幕逆向分析·a-bogus·x-gnarly
Blossom.1182 小时前
用一张“冰裂纹”石墨烯薄膜,让被动散热也能做 AI 推理——基于亚波长裂纹等离激元的零功耗温度-逻辑门
人工智能·深度学习·神经网络·目标检测·机器学习·机器人·语音识别