README.md 4.16 KB
Newer Older
zhangwenwei's avatar
zhangwenwei committed
1
2
# MMDetection3D

zhangwenwei's avatar
zhangwenwei committed
3
4
5
**News**: We released the codebase v0.1.0.

Documentation: https://mmdetection3d.readthedocs.io/
zhangwenwei's avatar
zhangwenwei committed
6
7
8

## Introduction

zhangwenwei's avatar
zhangwenwei committed
9
The master branch works with **PyTorch 1.3 to 1.5**.
zhangwenwei's avatar
zhangwenwei committed
10

zhangwenwei's avatar
zhangwenwei committed
11
12
MMDetection3D is an open source object detection toolbox based on PyTorch. It is
a part of the OpenMMLab project developed by [MMLab](http://mmlab.ie.cuhk.edu.hk/).
zhangwenwei's avatar
zhangwenwei committed
13

zhangwenwei's avatar
zhangwenwei committed
14
![demo image](demo/coco_test_12510.jpg)
zhangwenwei's avatar
zhangwenwei committed
15
16
17

### Major features

zhangwenwei's avatar
zhangwenwei committed
18
19
20
21
22
23
24
25
26
27
28
- **Modular Design**

  We decompose the detection framework into different components and one can easily construct a customized object detection framework by combining different modules.

- **Support of multiple frameworks out of box**

  The toolbox directly supports popular and contemporary detection frameworks, *e.g.* Faster RCNN, Mask RCNN, RetinaNet, etc.

- **High efficiency**

  The training speed is [faster than other codebases](./docs/benchmarks.md).
zhangwenwei's avatar
zhangwenwei committed
29

zhangwenwei's avatar
zhangwenwei committed
30
31
32
33
34
- **State of the art**

  The accuracy of models is [faster than other codebases](./docs/benchmarks.md).

Apart from MMDetection3D, we also released a library [MMDetection](https://github.com/open-mmlab/mmdetection) and [mmcv](https://github.com/open-mmlab/mmcv) for computer vision research, which are heavily depended on by this toolbox.
zhangwenwei's avatar
zhangwenwei committed
35
36
37
38
39

## License

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

zhangwenwei's avatar
zhangwenwei committed
40
## Changelog
zhangwenwei's avatar
zhangwenwei committed
41

zhangwenwei's avatar
zhangwenwei committed
42
43
v0.1.0 was released in 24/6/2020.
Please refer to [changelog.md](docs/changelog.md) for details and release history.
zhangwenwei's avatar
zhangwenwei committed
44
45
46
47

## Benchmark and model zoo

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

zhangwenwei's avatar
zhangwenwei committed
50
51
52
53
54
55
56
57
58
59
60
61
|                    | ResNet   | ResNeXt  | SENet    |PointNet++ | HRNet | RegNetX | Res2Net |
|--------------------|:--------:|:--------:|:--------:|:---------:|:-----:|:--------:|:-----:|
| SECOND             | ☐        | ☐        | ☐        | ✗         | ✓     | ✓        | ☐     |
| PointPillars       | ☐        | ☐        | ☐        | ✗         | ✓     | ✓        | ☐     |
| VoteNet            | ✗        | ✗        | ✗        | ✓         | ✗     | ✗        | ✗     |
| Part-A2            | ☐        | ☐        | ☐        | ✗         | ✓     | ✓        | ☐     |
| MVXNet             | ☐        | ☐        | ☐        | ✗         | ✓     | ✓        | ☐     |

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

**Notice**: All the models or modules supported in [MMDetection's model zoo](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
62
63
64

## Installation

zhangwenwei's avatar
zhangwenwei committed
65
Please refer to [install.md](docs/install.md) for installation and dataset preparation.
zhangwenwei's avatar
zhangwenwei committed
66
67
68
69


## Get Started

zhangwenwei's avatar
zhangwenwei committed
70
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
71
72
73

## Contributing

zhangwenwei's avatar
zhangwenwei committed
74
We appreciate all contributions to improve MMDetection. Please refer to [CONTRIBUTING.md](.github/CONTRIBUTING.md) for the contributing guideline.
zhangwenwei's avatar
zhangwenwei committed
75
76
77
78

## Acknowledgement

MMDetection3D is an open source project that is contributed by researchers and engineers from various colleges and companies. We appreciate all the contributors who implement their methods or add new features, as well as users who give valuable feedbacks.
zhangwenwei's avatar
zhangwenwei committed
79
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.
zhangwenwei's avatar
zhangwenwei committed
80
81
82
83


## Citation

zhangwenwei's avatar
zhangwenwei committed
84
85
86
87
88
89
90
91
92
93
94
95
If you use this toolbox or benchmark in your research, please cite this project.

```
@misc{mmdetection3d_2020,
  title   = {{MMDetection3D}},
  author  = {Zhang, Wenwei and Wu, Yuefeng and Li, Yinhao and Lin, Kwan-Yee and
             Qian, Chen, Shi, Jianping, and Chen, Kai, and Li, Hongsheng and
             Lin, Dahua, and Loy, Chen Change},
  howpublished = {\url{https://github.com/open-mmlab/mmdetection3d}},
  year =         {2020}
}
```
zhangwenwei's avatar
zhangwenwei committed
96
97
98
99


## Contact

zhangwenwei's avatar
zhangwenwei committed
100
This repo is currently maintained by Wenwei Zhang ([@ZwwWayne](https://github.com/ZwwWayne)).