
mmcv
OpenMMLab 通用视觉底座,算子与训练流程开箱即用
项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整
数据来源:GitHub API 逐日快照 · 收录于 2026-10-26 · 当前星标 ★ 6482
使用教程
环境要求
- 已安装 Python 环境(建议使用独立的虚拟环境)
- 已按 PyTorch 官方安装指南成功安装 PyTorch
- 可用的 pip 包管理工具
- 如需 CUDA 算子支持,环境中需有可用的 CUDA 与对应 PyTorch 版本
- Apple Silicon 用户需使用 PyTorch 1.13 及以上版本
安装与启动步骤
-
1安装 PyTorch
mmcv 依赖 PyTorch,需先按 PyTorch 官方安装指南装好与本机 CUDA 匹配的 PyTorch,再继续下一步。
-
2安装 openmim
openmim 是 OpenMMLab 的包管理工具,用它安装可自动匹配与当前 PyTorch/CUDA 对应的预编译包。
pip install -U openmim -
3安装完整版 mmcv
完整版包含各类 CUDA 算子,构建时间较长;有 CUDA 环境时官方推荐安装此版本。
mim install mmcv -
4安装指定版本
需要固定版本时在包名后加版本号,例如安装 2.0.0,便于与他人环境保持一致。
mim install mmcv==2.0.0 -
5安装轻量版
不需要 CUDA 算子时,可改装 mmcv-lite;注意它与完整版不能共存于同一环境。
pip install -U openmim mim install mmcv-lite -
6验证安装
进入 Python 交互环境执行 import mmcv,能正常导入且不报 ModuleNotFoundError 即安装成功。
如何确认成功
在 Python 中执行 import mmcv 无报错、可正常输出版本信息,即表示安装成功。
常见问题
Q:mmcv 和 mmcv-lite 有什么区别?
A:mmcv 为完整版,含各类 CUDA 算子,构建较慢;mmcv-lite 不含 CUDA 算子但保留其他功能,适合不需要自定义算子的场景。
Q:可以同时安装两个版本吗?
A:不可以。同一环境同时安装会报 ModuleNotFound 一类错误,需先卸载其中一个再装另一个。
Q:安装时下载的是 .tar.gz 而不是 .whl 怎么办?
A:说明没有与当前 PyTorch、CUDA 或 mmcv 版本匹配的预编译包,此时会转为源码编译,可参考官方 build 文档。
Q:Apple Silicon 设备有什么要求?
A:需使用 PyTorch 1.13 及以上版本,否则可能出现兼容问题。
注意事项
- 完整版与轻量版互斥,安装前请先卸载已存在的另一个版本。
- 有 CUDA 环境时官方推荐安装完整版 mmcv,以使用全部算子能力。
- 安装耗时主要在编译环节,若只用到基础功能可选轻量版以节省时间。
- 更多安装方式与源码编译步骤请查阅 MMCV 官方 Installation 文档。
核心亮点
- 注册器加配置继承机制,模块替换和实验管理成本低
- 内置大量经优化的 CPU/CUDA 自定义算子,覆盖检测分割常用操作
- 统一 Runner 封装训练验证推理、分布式与混合精度,减少重复代码
不足之处
- 自定义算子需本地编译,对 CUDA 与编译环境版本匹配要求较高
- 版本迭代较快,与上层算法库存在版本耦合,升级需同步对齐
- 文档以英文为主,部分算子与配置项缺少细粒度说明
适用场景
- 搭建目标检测或图像分割训练流水线时复用数据增强与训练循环
- 在 OpenMMLab 系列算法库之间迁移模型与配置
- 为视觉模型开发自定义 CUDA 算子并集成到训练流程
替代项目
detectron2、timm、kornia
上一篇:langvio
下一篇:Metalhead.jl
同类项目推荐
dlib
开源
C++ 写机器学习,人脸检测和训练模型一把梭
A toolkit for making real world machine learning and data analysis applications in C···
Texo
开源
20M 超轻量模型,浏览器里秒转公式为 LaTeX。
A minimalist SOTA LaTeX OCR model with only 20M parameters, running in browser. Full···
D-FINE-seg
开源
检测加分割一网打尽,训练到 TensorRT 部署全流程
D-FINE-seg Object Detection and Segmentation Framework (Train, Export, Inference)
surya
开源
多语言文档一键 OCR,版面表格阅读顺序全搞定
OCR, layout analysis, reading order, table recognition in 90+ languages
ovam
开源
用扩散模型生成精准的开放词汇语义分割图
Code for the paper Open-Vocabulary Attention Maps with Token Optimization for Semant···
fashn-vton-1.5
开源
上传衣服和照片,秒出试穿效果,无需手动抠图
FASHN VTON v1.5: Efficient Maskless Virtual Try-On in Pixel Space
ComfyUI-Qwen-VL-API
开源
在ComfyUI里直接调用通义千问视觉模型,让图像生成更智能
QWen-VL-Plus & QWen-VL-Max in ComfyUI
3DGS-DET
开源
用3D高斯泼溅做室内3D检测,边界更准、小目标更稳
Official codes for paper: 3DGS-DET: Empower 3D Gaussian Splatting with Boundary Gu···
awesome-yolo-object-detection
开源
一份清单,快速找到你想要的 YOLO 模型和数据集。
A collection of some awesome public YOLO object detection series projects and the re···