从零开始搭建并运行mmsegmentation

安装:

Step 1:创建Conda 环境并激活之

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

Step 2:CUDA版本选择,及安装Pytorch

关于设备GPU的cuda版本,根据如下的选择原则:

  • 对于Ampere架构的NVIDIA的GPU,例如GeForce 30系列核NVIDIA A100,必须安装CUDA11。
  • 对于旧版的NVIDIA GPUS,CUDA 11 是向下兼容的,但是 CUDA10.2会更轻量化并且表现更好的性能

确定好cuda版本后,进入Torch官网选择对应版本的Pytorch进行安装

Step 3:安装 MMCV using MIM.

复制代码
pip install -U openmim
mim install mmengine
mim install "mmcv>=2.0.0"

Step 4:安装 MMSegmentation.

复制代码
git clone -b main https://github.com/open-mmlab/mmsegmentation.git
cd mmsegmentation
pip install -v -e .
# '-v' means verbose, or more output
# '-e' means installing a project in editable mode,
# thus any local modifications made to the code will take effect without reinstallation.

验证:

Step 1. We need to download config and checkpoint files.下载config和checkpoint 文件

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

Step 2. Verify the inference demo.验证推理demo

Option (a). If you install mmsegmentation from source, just run the following command.

方式(a)如果你从源码安装mmsegmentation,只需要运行下面的命令

复制代码
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

You will see a new image result.jpg on your current folder, where segmentation masks are covered on all objects.

你如果看到了一个新图片result.jpg在你当前文件夹,说明安装成功了。(运行时,可能会有一些warning,但不影响结果的生成)

相关推荐
一 铭33 分钟前
AI领域新趋势:从提示(Prompt)工程到上下文(Context)工程
人工智能·语言模型·大模型·llm·prompt
麻雀无能为力4 小时前
CAU数据挖掘实验 表分析数据插件
人工智能·数据挖掘·中国农业大学
时序之心4 小时前
时空数据挖掘五大革新方向详解篇!
人工智能·数据挖掘·论文·时间序列
.30-06Springfield5 小时前
人工智能概念之七:集成学习思想(Bagging、Boosting、Stacking)
人工智能·算法·机器学习·集成学习
说私域6 小时前
基于开源AI智能名片链动2+1模式S2B2C商城小程序的超级文化符号构建路径研究
人工智能·小程序·开源
永洪科技6 小时前
永洪科技荣获商业智能品牌影响力奖,全力打造”AI+决策”引擎
大数据·人工智能·科技·数据分析·数据可视化·bi
shangyingying_16 小时前
关于小波降噪、小波增强、小波去雾的原理区分
人工智能·深度学习·计算机视觉
书玮嘎7 小时前
【WIP】【VLA&VLM——InternVL系列】
人工智能·深度学习
猫头虎7 小时前
猫头虎 AI工具分享:一个网页抓取、结构化数据提取、网页爬取、浏览器自动化操作工具:Hyperbrowser MCP
运维·人工智能·gpt·开源·自动化·文心一言·ai编程
要努力啊啊啊7 小时前
YOLOv2 正负样本分配机制详解
人工智能·深度学习·yolo·计算机视觉·目标跟踪