README.md 5.18 KB
Newer Older
zhangwenwei's avatar
zhangwenwei committed
1
<div align="center">
zhangwenwei's avatar
zhangwenwei committed
2
  <img src="resources/mmdet3d-logo.png" width="600"/>
zhangwenwei's avatar
zhangwenwei committed
3
</div>
zhangwenwei's avatar
zhangwenwei committed
4

Wenwei Zhang's avatar
Wenwei Zhang committed
5
6
7
8
9
10
[![docs](https://img.shields.io/badge/docs-latest-blue)](https://mmdetection3d.readthedocs.io/en/latest/)
[![badge](https://github.com/open-mmlab/mmdetection3d/workflows/build/badge.svg)](https://github.com/open-mmlab/mmdetection3d/actions)
[![codecov](https://codecov.io/gh/open-mmlab/mmdetection3d/branch/master/graph/badge.svg)](https://codecov.io/gh/open-mmlab/mmdetection3d)
[![license](https://img.shields.io/github/license/open-mmlab/mmdetection3d.svg)](https://github.com/open-mmlab/mmdetection3d/blob/master/LICENSE)


zhangwenwei's avatar
zhangwenwei committed
11
12
13
**News**: We released the codebase v0.1.0.

Documentation: https://mmdetection3d.readthedocs.io/
zhangwenwei's avatar
zhangwenwei committed
14
15
16

## Introduction

Wenwei Zhang's avatar
Wenwei Zhang committed
17
The master branch works with **PyTorch 1.3 to 1.6**.
zhangwenwei's avatar
zhangwenwei committed
18

19
MMDetection3D is an open source object detection toolbox based on PyTorch, towards the next-generation platform for general 3D detection. It is
zhangwenwei's avatar
zhangwenwei committed
20
a part of the OpenMMLab project developed by [MMLab](http://mmlab.ie.cuhk.edu.hk/).
zhangwenwei's avatar
zhangwenwei committed
21

zhangwenwei's avatar
zhangwenwei committed
22
![demo image](resources/mmdet3d_outdoor_demo.gif)
zhangwenwei's avatar
zhangwenwei committed
23
24
25

### Major features

zhangwenwei's avatar
zhangwenwei committed
26
- **Support multi-modality/single-modality detectors out of box**
zhangwenwei's avatar
zhangwenwei committed
27

28
  It directly supports multi-modality/single-modality detectors including MVXNet, VoteNet, PointPillars, etc.
zhangwenwei's avatar
zhangwenwei committed
29

zhangwenwei's avatar
zhangwenwei committed
30
- **Support indoor/outdoor 3D detection out of box**
zhangwenwei's avatar
zhangwenwei committed
31

32
  It directly supports popular indoor and outdoor 3D detection datasets, including ScanNet, SUNRGB-D, nuScenes, Lyft, and KITTI.
zhangwenwei's avatar
zhangwenwei committed
33

zhangwenwei's avatar
zhangwenwei committed
34
- **Natural integration with 2D detection**
35

Wenwei Zhang's avatar
Wenwei Zhang committed
36
  All the about **40+ methods, 300+ models**, and modules supported in [MMDetection](https://github.com/open-mmlab/mmdetection/blob/master/docs/model_zoo.md) can be trained or used in this codebase.
zhangwenwei's avatar
zhangwenwei committed
37

zhangwenwei's avatar
zhangwenwei committed
38
- **High efficiency**
zhangwenwei's avatar
zhangwenwei committed
39

Wenwei Zhang's avatar
Wenwei Zhang committed
40
  It trains faster than other codebases. The main results are as below. Details can be found in [benchmark.md](./docs/benchmarks.md). We compare the number of samples trained per second (the higher, the better). The models that are not supported by other codebases are marked by `×`.
zhangwenwei's avatar
zhangwenwei committed
41

Wenwei Zhang's avatar
Wenwei Zhang committed
42
43
44
45
46
47
48
49
50
  | Methods | MMDetection3D | [OpenPCDet](https://github.com/open-mmlab/OpenPCDet) |[votenet](https://github.com/facebookresearch/votenet)| [Det3D](https://github.com/poodarchu/Det3D) |
  |:-------:|:-------------:|:---------:|:-----:|:-----:|
  | VoteNet | 358           | ×         |   77  | ×     |
  | PointPillars-car| 141           | ×         |   ×  | 140     |
  | PointPillars-3class| 107           |44     |   ×      | ×    |
  | SECOND| 40           |30     |   ×      | ×    |
  | Part-A2| 17           |14     |   ×      | ×    |

Like [MMDetection](https://github.com/open-mmlab/mmdetection) and [MMCV](https://github.com/open-mmlab/mmcv), MMDetection3D can also be used as a library to support different projects on top of it.
zhangwenwei's avatar
zhangwenwei committed
51
52
53
54
55

## License

This project is released under the [Apache 2.0 license](LICENSE).

zhangwenwei's avatar
zhangwenwei committed
56
## Changelog
zhangwenwei's avatar
zhangwenwei committed
57

58
v0.1.0 was released in 9/7/2020.
zhangwenwei's avatar
zhangwenwei committed
59
Please refer to [changelog.md](docs/changelog.md) for details and release history.
zhangwenwei's avatar
zhangwenwei committed
60
61
62
63

## Benchmark and model zoo

Supported methods and backbones are shown in the below table.
zhangwenwei's avatar
zhangwenwei committed
64
Results and models are available in the [model zoo](docs/model_zoo.md).
zhangwenwei's avatar
zhangwenwei committed
65

zhangwenwei's avatar
zhangwenwei committed
66
67
|                    | ResNet   | ResNeXt  | SENet    |PointNet++ | HRNet | RegNetX | Res2Net |
|--------------------|:--------:|:--------:|:--------:|:---------:|:-----:|:--------:|:-----:|
68
69
70
| SECOND             | ☐        | ☐        | ☐        | ✗         | ☐     | ✓        | ☐     |
| PointPillars       | ☐        | ☐        | ☐        | ✗         | ☐     | ✓        | ☐     |
| FreeAnchor         | ☐        | ☐        | ☐        | ✗         | ☐     | ✓        | ☐     |
zhangwenwei's avatar
zhangwenwei committed
71
| VoteNet            | ✗        | ✗        | ✗        | ✓         | ✗     | ✗        | ✗     |
72
73
| Part-A2            | ☐        | ☐        | ☐        | ✗         | ☐     | ✓        | ☐     |
| MVXNet             | ☐        | ☐        | ☐        | ✗         | ☐     | ✓        | ☐     |
zhangwenwei's avatar
zhangwenwei committed
74
75
76
77

Other features
- [x] [Dynamic Voxelization](configs/carafe/README.md)

78
**Note:** All the about **300 models, methods of 40+ papers** in 2D detection supported by [MMDetection](https://github.com/open-mmlab/mmdetection/blob/master/docs/model_zoo.md) can be trained or used in this codebase.
zhangwenwei's avatar
zhangwenwei committed
79
80
81

## Installation

zhangwenwei's avatar
zhangwenwei committed
82
Please refer to [install.md](docs/install.md) for installation and dataset preparation.
zhangwenwei's avatar
zhangwenwei committed
83
84
85

## Get Started

zhangwenwei's avatar
zhangwenwei committed
86
Please see [getting_started.md](docs/getting_started.md) for the basic usage of MMDetection. There are also tutorials for [finetuning models](docs/tutorials/finetune.md), [adding new dataset](docs/tutorials/new_dataset.md), [designing data pipeline](docs/tutorials/data_pipeline.md), and [adding new modules](docs/tutorials/new_modules.md).
zhangwenwei's avatar
zhangwenwei committed
87
88
89

## Contributing

zhangwenwei's avatar
zhangwenwei committed
90
We appreciate all contributions to improve MMDetection3D. Please refer to [CONTRIBUTING.md](.github/CONTRIBUTING.md) for the contributing guideline.
zhangwenwei's avatar
zhangwenwei committed
91
92
93

## Acknowledgement

zhangwenwei's avatar
zhangwenwei committed
94
MMDetection3D is an open source project that is contributed by researchers and engineers from various colleges and companies. We appreciate all the contributors as well as users who give valuable feedbacks.
zhangwenwei's avatar
zhangwenwei committed
95
We wish that the toolbox and benchmark could serve the growing research community by providing a flexible toolkit to reimplement existing methods and develop their own new 3D detectors.