安装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
相关推荐
那雨倾城18 小时前
PiscCode轨迹跟踪Mediapipe + OpenCV进阶:速度估算
图像处理·人工智能·python·opencv·计算机视觉
2501_9200470318 小时前
bash自带的切片操作
开发语言·python·bash
偷心伊普西隆18 小时前
Python EXCEL 小技巧:最快重新排列dataframe函数
python·excel
兮希yxx18 小时前
conda配置pytorch虚拟环境
人工智能·pytorch·conda
我是海飞19 小时前
Tensorflow Lite 的yes/no语音识别音频预处理模型训练教程
python·学习·tensorflow·音视频·嵌入式·语音识别
know__ledge19 小时前
Python学习3.0使用Unittest框架运行测试用例
python·学习·测试用例
豆浩宇19 小时前
学习PaddlePaddle--环境配置-Windows 11 + RTX 4060
人工智能·windows·深度学习·学习·目标检测·计算机视觉·paddlepaddle
qq_3391911419 小时前
深度学习玩游戏, 模型玩游戏,大模型+游戏 llm+game, 机器学习玩游戏,人工智能游戏陪伴,模型陪玩游戏
人工智能·深度学习·玩游戏
鲸落落丶20 小时前
finalize() 方法介绍
开发语言·python
WangYan202220 小时前
MATLAB 2023a深度学习工具箱全面解析:从CNN、RNN、GAN到YOLO与U-Net,涵盖模型解释、迁移学习、时间序列预测与图像生成的完整实战指南
深度学习·matlab·matlab 2023a