MMCV is a foundational python library for computer vision research and supports many
MMCV is a foundational python library for computer vision research and supports many
research projects in MMLAB as below:
research projects as below:
-[MMDetection](https://github.com/open-mmlab/mmdetection): Detection toolbox and benchmark
-[MMDetection](https://github.com/open-mmlab/mmdetection): Detection toolbox and benchmark
-[MMDetection3D](https://github.com/open-mmlab/mmdetection3d): General 3D object detection toolbox and benchmark
-[MMDetection3D](https://github.com/open-mmlab/mmdetection3d): General 3D object detection toolbox and benchmark
...
@@ -38,6 +38,8 @@ There are two versions of MMCV:
...
@@ -38,6 +38,8 @@ 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**: 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.
-**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
### Install with pip
a. Install the lite version.
a. Install the lite version.
...
@@ -48,7 +50,9 @@ pip install mmcv
...
@@ -48,7 +50,9 @@ pip install mmcv
b. Install the full version.
b. Install the full version.
We provide the pre-built mmcv package with different PyTorch and CUDA versions to simplify the building.
Before installing mmcv-full, make sure that PyTorch has been successfully installed following the [official guide](https://pytorch.org/).
We provide pre-built mmcv packages (recommended) with different PyTorch and CUDA versions to simplify the building.
<tableclass="docutils"><tbody><tr><thwidth="80"> CUDA </th><thvalign="bottom"align="left"width="100">torch 1.5</th><thvalign="bottom"align="left"width="100">torch 1.4</th><thvalign="bottom"align="left"width="100">torch 1.3</th></tr>
<tableclass="docutils"><tbody><tr><thwidth="80"> CUDA </th><thvalign="bottom"align="left"width="100">torch 1.5</th><thvalign="bottom"align="left"width="100">torch 1.4</th><thvalign="bottom"align="left"width="100">torch 1.3</th></tr>
@@ -117,7 +118,7 @@ An example json file could be like:
...
@@ -117,7 +118,7 @@ An example json file could be like:
}
}
```
```
The default links of the pre-trained models hosted on Open-MMLab AWS could be found [here](../mmcv/model_zoo/open_mmlab.json).
The default links of the pre-trained models hosted on OpenMMLab AWS could be found [here](https://github.com/open-mmlab/mmcv/blob/master/mmcv/model_zoo/open_mmlab.json).
You may override default links by putting `open-mmlab.json` under `MMCV_HOME`. If `MMCV_HOME` is not find in the environment, `~/.cache/mmcv` will be used by default. You may `export MMCV_HOME=/your/path` to use your own path.
You may override default links by putting `open-mmlab.json` under `MMCV_HOME`. If `MMCV_HOME` is not find in the environment, `~/.cache/mmcv` will be used by default. You may `export MMCV_HOME=/your/path` to use your own path.