mmcv

OpenMMLab 通用视觉底座,算子与训练流程开箱即用

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 计算机视觉 🍴 复刻 1770 🐛 待处理 465 👀 关注 84
GitHub ↗
GitHub 星标 ★ 6482
维护状态 较活跃
是否开源 开源
定价模式 🏷️完全免费 开源免费(Apache-2.0)

项目预览

mmcv 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

分类计算机视觉
开发团队open-mmlab
定价模式🏷️完全免费
价格说明开源免费(Apache-2.0)
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议Apache-2.0
主要语言Python
GitHub 星标★ 6482
上线时间2018-08-22 00:00:00
代码更新2026-09-28
仓库体积14.3 MB
本站数据更新2026-10-26
维护状态较活跃
收录时间2026-10-26
浏览次数0

项目介绍

MMCV 是 OpenMMLab 生态的计算机视觉基础库,为检测、分割、分类等上层算法框架提供统一底层能力。它解决视觉研究中算子实现分散、训练流程重复搭建、跨项目兼容性差的问题,把常用卷积与注意力算子、数据加载与增强流水线、模型构建与权重初始化、分布式训练与混合精度、日志与检查点管理等封装为可复用组件。核心能力包括:基于注册器机制的模块化构建,便于替换骨干网络与损失函数;丰富的 CPU/CUDA 自定义算子与算子编译工具;配置系统支持继承与变量引用,方便管理大规模实验;提供 Runner 统一训练验证推理流程,并配套可视化与基准测试工具。项目以 Python 为主,提供 C++/CUDA 扩展,是 OpenMMLab 各算法库的共同依赖。

Star 增长趋势

本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整

数据来源:GitHub API 逐日快照 · 收录于 2026-10-26 · 当前星标 ★ 6482

使用教程

难度:入门 约 10 分钟 部署方式:库/依赖 6 步

环境要求

  • 已安装 Python 环境(建议使用独立的虚拟环境)
  • 已按 PyTorch 官方安装指南成功安装 PyTorch
  • 可用的 pip 包管理工具
  • 如需 CUDA 算子支持,环境中需有可用的 CUDA 与对应 PyTorch 版本
  • Apple Silicon 用户需使用 PyTorch 1.13 及以上版本

安装与启动步骤

  1. 1安装 PyTorch

    mmcv 依赖 PyTorch,需先按 PyTorch 官方安装指南装好与本机 CUDA 匹配的 PyTorch,再继续下一步。

  2. 2安装 openmim

    openmim 是 OpenMMLab 的包管理工具,用它安装可自动匹配与当前 PyTorch/CUDA 对应的预编译包。

    pip install -U openmim
  3. 3安装完整版 mmcv

    完整版包含各类 CUDA 算子,构建时间较长;有 CUDA 环境时官方推荐安装此版本。

    mim install mmcv
  4. 4安装指定版本

    需要固定版本时在包名后加版本号,例如安装 2.0.0,便于与他人环境保持一致。

    mim install mmcv==2.0.0
  5. 5安装轻量版

    不需要 CUDA 算子时,可改装 mmcv-lite;注意它与完整版不能共存于同一环境。

    pip install -U openmim
    mim install mmcv-lite
  6. 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···

★ 14465 2026-08-10
Texo 开源

20M 超轻量模型,浏览器里秒转公式为 LaTeX。

A minimalist SOTA LaTeX OCR model with only 20M parameters, running in browser. Full···

★ 910 2026-09-29
D-FINE-seg 开源

检测加分割一网打尽,训练到 TensorRT 部署全流程

D-FINE-seg Object Detection and Segmentation Framework (Train, Export, Inference)

★ 219 2026-09-20
surya 开源

多语言文档一键 OCR,版面表格阅读顺序全搞定

OCR, layout analysis, reading order, table recognition in 90+ languages

★ 21534 2026-09-19
ovam 开源

用扩散模型生成精准的开放词汇语义分割图

Code for the paper Open-Vocabulary Attention Maps with Token Optimization for Semant···

★ 70 2026-08-09
fashn-vton-1.5 开源

上传衣服和照片,秒出试穿效果,无需手动抠图

FASHN VTON v1.5: Efficient Maskless Virtual Try-On in Pixel Space

★ 364 2026-08-09
ComfyUI-Qwen-VL-API 开源

在ComfyUI里直接调用通义千问视觉模型,让图像生成更智能

QWen-VL-Plus & QWen-VL-Max in ComfyUI

★ 220 2026-08-09
3DGS-DET 开源

用3D高斯泼溅做室内3D检测,边界更准、小目标更稳

Official codes for paper: 3DGS-DET: Empower 3D Gaussian Splatting with Boundary Gu···

★ 164 2026-09-30
awesome-yolo-object-detection 开源

一份清单,快速找到你想要的 YOLO 模型和数据集。

A collection of some awesome public YOLO object detection series projects and the re···

★ 1794 2026-08-09