You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
wang shiguang 23ecd83d59
Add parrots ci (#662)
9 months ago
.github/workflows Add parrots ci (#662) 9 months ago
docs Add documentations for registry (#656) 10 months ago
examples Fix docstring formats (#383) 1 year ago
mmcv [Enhance] Refactor logger (#659) 10 months ago
requirements Support pytorch 1.7 and update the CI (#631) 10 months ago
tests [Enhance] Refactor logger (#659) 10 months ago
.dockerignore fix dependencies (#146) 1 year ago
.gitignore Add docs for CNN modules (#266) 1 year ago
.pre-commit-config.yaml lint markdown files (#592) 11 months ago
.readthedocs.yml Support pytorch 1.7 and update the CI (#631) 10 months ago
CONTRIBUTING.md lint markdown files (#592) 11 months ago
Dockerfile fix dependencies (#146) 1 year ago
Jenkinsfile Update jenkinsfile (#444) 1 year ago
LICENSE Add copyright header (#171) 1 year ago
MANIFEST.in Support pytorch 1.7 and update the CI (#631) 10 months ago
README.md update README.md for torch 1.7 installation (#642) 10 months ago
requirements.txt Replace `re` with `regex` (#457) 1 year ago
setup.cfg add unittest for onnx convert (#608) 10 months ago
setup.py Support pytorch 1.7 and update the CI (#631) 10 months ago

README.md

PyPI badge codecov license

Introduction

MMCV is a foundational python library for computer vision research and supports many
research projects as below:

It provides the following functionalities.

  • Universal IO APIs
  • Image/Video processing
  • Image and annotation visualization
  • Useful utilities (progress bar, timer, ...)
  • PyTorch runner with hooking mechanism
  • Various CNN architectures
  • High-quality implementation of common CUDA ops

See the documentation for more features and usage.

Note: MMCV requires Python 3.6+.

Installation

There are two versions of MMCV:

  • mmcv: lite, without CUDA ops but all other features, similar to mmcv<1.0.0. It is useful when you do not need those CUDA ops.
  • mmcv-full: comprehensive, with full features and various CUDA ops out of box. It takes longer time to build.

Note: Do not install both versions in the same environment, otherwise you may encounter errors like ModuleNotFound. You need to uninstall one before installing the other.

Install with pip

a. Install the lite version.

pip install mmcv

b. Install the full version.

Before installing mmcv-full, make sure that PyTorch has been successfully installed following the official guide.

We provide pre-built mmcv packages (recommended) with different PyTorch and CUDA versions to simplify the building.

CUDA torch 1.7 torch 1.6 torch 1.5 torch 1.4 torch 1.3
11.0
install
pip install mmcv-full==latest+torch1.7.0+cu110 -f https://download.openmmlab.com/mmcv/dist/index.html
10.2
install
pip install mmcv-full==latest+torch1.7.0+cu102 -f https://download.openmmlab.com/mmcv/dist/index.html
install
pip install mmcv-full==latest+torch1.6.0+cu102 -f https://download.openmmlab.com/mmcv/dist/index.html
install
pip install mmcv-full==latest+torch1.5.0+cu102 -f https://download.openmmlab.com/mmcv/dist/index.html
10.1
install
 pip install mmcv-full==latest+torch1.7.0+cu101 -f https://download.openmmlab.com/mmcv/dist/index.html
install
 pip install mmcv-full==latest+torch1.6.0+cu101 -f https://download.openmmlab.com/mmcv/dist/index.html
install
 pip install mmcv-full==latest+torch1.5.0+cu101 -f https://download.openmmlab.com/mmcv/dist/index.html
install
pip install mmcv-full==latest+torch1.4.0+cu101 -f https://download.openmmlab.com/mmcv/dist/index.html
install
pip install mmcv-full==latest+torch1.3.0+cu101 -f https://download.openmmlab.com/mmcv/dist/index.html
9.2
install
 pip install mmcv-full==latest+torch1.7.0+cu92 -f https://download.openmmlab.com/mmcv/dist/index.html
install
 pip install mmcv-full==latest+torch1.6.0+cu92 -f https://download.openmmlab.com/mmcv/dist/index.html
install
 pip install mmcv-full==latest+torch1.5.0+cu92 -f https://download.openmmlab.com/mmcv/dist/index.html
install
pip install mmcv-full==latest+torch1.4.0+cu92 -f https://download.openmmlab.com/mmcv/dist/index.html
install
pip install mmcv-full==latest+torch1.3.0+cu92 -f https://download.openmmlab.com/mmcv/dist/index.html
cpu
install
 pip install mmcv-full==latest+torch1.7.0+cpu -f https://download.openmmlab.com/mmcv/dist/index.html
install
 pip install mmcv-full==latest+torch1.6.0+cpu -f https://download.openmmlab.com/mmcv/dist/index.html
install
 pip install mmcv-full==latest+torch1.5.0+cpu -f https://download.openmmlab.com/mmcv/dist/index.html
install
pip install mmcv-full==latest+torch1.4.0+cpu -f https://download.openmmlab.com/mmcv/dist/index.html
install
pip install mmcv-full==latest+torch1.3.0+cpu -f https://download.openmmlab.com/mmcv/dist/index.html

Another way is to compile locally by running

pip install mmcv-full

Note that the local compiling may take up to 10 mins.

Install from source

After cloning the repo with

git clone https://github.com/open-mmlab/mmcv.git
cd mmcv

You can either

  • install the lite version

    pip install -e .
    
  • install the full version

    MMCV_WITH_OPS=1 pip install -e .
    

If you are on macOS, add the following environment variables before the installing command.

CC=clang CXX=clang++ CFLAGS='-stdlib=libc++'

e.g.,

CC=clang CXX=clang++ CFLAGS='-stdlib=libc++' MMCV_WITH_OPS=1 pip install -e .

If you are on Windows10, set the following environment variable before the installing command.

set MMCV_WITH_OPS=1

e.g.,

set MMCV_WITH_OPS=1
pip install -e .

Note: If you would like to use opencv-python-headless instead of opencv-python,
e.g., in a minimum container environment or servers without GUI,
you can first install it before installing MMCV to skip the installation of opencv-python.

TroubleShooting

If you meet issues when running or compiling mmcv, we list some common issues in TROUBLESHOOTING.md.

简介

MM系列基础架构

ai开发工具

Python Cuda C++ Pickle other

贡献者 (82)

chenkaidev@gmail.com xvjiarui0826@gmail.com 40779233+ZwwWayne@users.noreply.github.com yhcao6@gmail.com wangxinjiang@sensetime.com hejm37@mail2.sysu.edu.cn innerlee@users.noreply.github.com linjintao@sensetime.com pjmzju@gmail.com xiongyuxy@gmail.com yl93528@gmail.com 34728304+JoannaLXY@users.noreply.github.com nbeiray@gmail.com 34116221+v-qjqs@users.noreply.github.com drcut@users.noreply.github.com daviddelaiglesiacastro@gmail.com 528557675@qq.com lindahua@gmail.com xusu@sensetime.com guwang12@gmail.com 31604046+hanyc0914@users.noreply.github.com guwang12@foxmail.com pangjiangmiao@gmail.com maxim@bonnaerens.be xlide@connect.ust.hk thangvubk@gmail.com xiaojieli@buaa.edu.cn thuzhaoyue@gmail.com 515740295@qq.com 49829199+yamengxi@users.noreply.github.com karlind@users.noreply.github.com qiwen@qiwen.name miguelmndez@gmail.com xurui4@sensetime.com 12756472+wangruohui@users.noreply.github.com swanxinjiang@gmail.com wxt1994@126.com 790230517@qq.com zhuyuanhao@sensetime.com hanzhipeng9@gmail.com 67114543+louzana@users.noreply.github.com streetyao@live.com 61961338+jshilong@users.noreply.github.com 26602940+0xflotus@users.noreply.github.com 965813422@qq.com 46709554+z-bingo@users.noreply.github.com 58807745+ChaseMonsterAway@users.noreply.github.com Danil328@yandex.ru diglesia@gradiant.org 31407075+dhananjaisharma10@users.noreply.github.com nizhibitsky@ya.ru gongtao950513@gmail.com 34324155+kennymckormick@users.noreply.github.com 1052020748@qq.com 506365608@qq.com GodBlessZhk@outlook.com 40995042+ChiangYintso@users.noreply.github.com 31381602+johnzja@users.noreply.github.com nesoriti@yandex.ru oozorakuro409@gmail.com matt.dawkins@kitware.com matthew.d.dawkins@gmail.com qrnnis2623891@gmail.com 849906763@qq.com rinatshigapov@gmail.com ternaus@users.noreply.github.com lfc199471@gmail.com 879316283@qq.com yeliudev@gmail.com this@zyc.ai kemoho@stu.xmu.edu.cn zhanzy5@mail2.sysu.edu.cn 15834119171@163.com dongqxia@umich.edu wswday@sina.com tianyuandu@gmail.com tjsongzw@gmail.com wangshiguang@sensetime.com 38210459+wdmwhh@users.noreply.github.com youkaichao@126.com nju.zhuyuanhao@gmail.com