Based on version mmdetection3d-0.17.1(环境配置好之后可运行).
- 【202212】mmdet3d-0.17版本环境配置(CUDA 11.x + torch1.10.1)
- TODO:version mmdetection3d-1.1.
TODO:
- 【202212done】目标检测最新论文实时更新
- 2024语义分割最新论文实时更新
- 【202209done】目标检测框架(pcdet+mmdetection3d+det3d+paddle3d)文章撰写
- 【202406done】3D语义分割框架综述(mmdetection3d|OpenPCSeg|Pointcept)
- 数据集详细剖析:kitti&waymo&nuScenes
- Apollo学习https://github.com/HuangCongQing/apollo_note
Documentation: https://mmdetection3d.readthedocs.io/
学习文档:https://www.yuque.com/huangzhongqing/hre6tf/nnioxg
代码注解
-
模型配置注释:
- 配置示例1:votenet.py示例代码(base_/models/):votenet.py
- 配置示例2:pointpillars.py:configs/base/models/hv_pointpillars_secfpn_ouster.py
-
kitti评测详细介绍(可适配自己的数据集评测):eval.py
其他目标检测框架(pcdet+mmdetection3d+det3d+paddle3d)代码注解笔记:
- pcdet:https://github.com/HuangCongQing/pcdet-note
- mmdetection3d:https://github.com/HuangCongQing/mmdetection3d-note
- det3d: TODO
- paddle3dL TODO
创建一个知识星球 【自动驾驶感知(PCL/ROS+DL)】 专注于自动驾驶感知领域,包括传统方法(PCL点云库,ROS)和深度学习(目标检测+语义分割)方法。同时涉及Apollo,Autoware(基于ros2),BEV感知,三维重建,SLAM(视觉+激光雷达) ,模型压缩(蒸馏+剪枝+量化等),自动驾驶模拟仿真,自动驾驶数据集标注&数据闭环等自动驾驶全栈技术,欢迎扫码二维码加入,一起登顶自动驾驶的高峰!
News: We released the codebase v0.17.0.
In the nuScenes 3D detection challenge of the 5th AI Driving Olympics in NeurIPS 2020, we obtained the best PKL award and the second runner-up by multi-modality entry, and the best vision-only results.
Code and models for the best vision-only method, FCOS3D, have been released. Please stay tuned for MoCa.
English | 简体中文
The master branch works with PyTorch 1.3+.
MMDetection3D is an open source object detection toolbox based on PyTorch, towards the next-generation platform for general 3D detection. It is a part of the OpenMMLab project developed by MMLab.
-
Support multi-modality/single-modality detectors out of box
It directly supports multi-modality/single-modality detectors including MVXNet, VoteNet, PointPillars, etc.
-
Support indoor/outdoor 3D detection out of box
It directly supports popular indoor and outdoor 3D detection datasets, including ScanNet, SUNRGB-D, Waymo, nuScenes, Lyft, and KITTI. For nuScenes dataset, we also support nuImages dataset.
-
Natural integration with 2D detection
All the about 300+ models, methods of 40+ papers, and modules supported in MMDetection can be trained or used in this codebase.
-
High efficiency
It trains faster than other codebases. The main results are as below. Details can be found in benchmark.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
×
.Methods MMDetection3D OpenPCDet votenet Det3D VoteNet 358 × 77 × PointPillars-car 141 × × 140 PointPillars-3class 107 44 × × SECOND 40 30 × × Part-A2 17 14 × ×
Like MMDetection and MMCV, MMDetection3D can also be used as a library to support different projects on top of it.
This project is released under the Apache 2.0 license.
v0.17.0 was released in 1/9/2021. Please refer to changelog.md for details and release history.
Supported methods and backbones are shown in the below table. Results and models are available in the model zoo.
Support backbones:
- PointNet (CVPR'2017)
- PointNet++ (NeurIPS'2017)
- RegNet (CVPR'2020)
Support methods
- SECOND (Sensor'2018)
- PointPillars (CVPR'2019)
- FreeAnchor (NeurIPS'2019)
- VoteNet (ICCV'2019)
- H3DNet (ECCV'2020)
- 3DSSD (CVPR'2020)
- Part-A2 (TPAMI'2020)
- MVXNet (ICRA'2019)
- CenterPoint (CVPR'2021)
- SSN (ECCV'2020)
- ImVoteNet (CVPR'2020)
- FCOS3D (Arxiv'2021)
- PointNet++ (NeurIPS'2017)
- Group-Free-3D (Arxiv'2021)
- ImVoxelNet (Arxiv'2021)
- PAConv (CVPR'2021)
ResNet | ResNeXt | SENet | PointNet++ | HRNet | RegNetX | Res2Net | |
---|---|---|---|---|---|---|---|
SECOND | ☐ | ☐ | ☐ | ✗ | ☐ | ✓ | ☐ |
PointPillars | ☐ | ☐ | ☐ | ✗ | ☐ | ✓ | ☐ |
FreeAnchor | ☐ | ☐ | ☐ | ✗ | ☐ | ✓ | ☐ |
VoteNet | ✗ | ✗ | ✗ | ✓ | ✗ | ✗ | ✗ |
H3DNet | ✗ | ✗ | ✗ | ✓ | ✗ | ✗ | ✗ |
3DSSD | ✗ | ✗ | ✗ | ✓ | ✗ | ✗ | ✗ |
Part-A2 | ☐ | ☐ | ☐ | ✗ | ☐ | ✓ | ☐ |
MVXNet | ☐ | ☐ | ☐ | ✗ | ☐ | ✓ | ☐ |
CenterPoint | ☐ | ☐ | ☐ | ✗ | ☐ | ✓ | ☐ |
SSN | ☐ | ☐ | ☐ | ✗ | ☐ | ✓ | ☐ |
ImVoteNet | ✗ | ✗ | ✗ | ✓ | ✗ | ✗ | ✗ |
FCOS3D | ✓ | ☐ | ☐ | ✗ | ☐ | ☐ | ☐ |
PointNet++ | ✗ | ✗ | ✗ | ✓ | ✗ | ✗ | ✗ |
Group-Free-3D | ✗ | ✗ | ✗ | ✓ | ✗ | ✗ | ✗ |
ImVoxelNet | ✓ | ✗ | ✗ | ✗ | ✗ | ✗ | ✗ |
PAConv | ✗ | ✗ | ✗ | ✓ | ✗ | ✗ | ✗ |
Other features
Note: All the about 300+ models, methods of 40+ papers in 2D detection supported by MMDetection can be trained or used in this codebase.
Please refer to getting_started.md for installation.
Please see getting_started.md for the basic usage of MMDetection3D. We provide guidance for quick run with existing dataset and with customized dataset for beginners. There are also tutorials for learning configuration systems, adding new dataset, designing data pipeline, customizing models, customizing runtime settings and Waymo dataset.
Please refer to FAQ for frequently asked questions. When updating the version of MMDetection3D, please also check the compatibility doc to be aware of the BC-breaking updates introduced in each version.
If you find this project useful in your research, please consider cite:
@misc{mmdet3d2020,
title={{MMDetection3D: OpenMMLab} next-generation platform for general {3D} object detection},
author={MMDetection3D Contributors},
howpublished = {\url{https://github.com/open-mmlab/mmdetection3d}},
year={2020}
}
We appreciate all contributions to improve MMDetection3D. Please refer to CONTRIBUTING.md for the contributing guideline.
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. 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.
- MMCV: OpenMMLab foundational library for computer vision.
- MIM: MIM Installs OpenMMLab Packages.
- MMClassification: OpenMMLab image classification toolbox and benchmark.
- MMDetection: OpenMMLab detection toolbox and benchmark.
- MMDetection3D: OpenMMLab next-generation platform for general 3D object detection.
- MMSegmentation: OpenMMLab semantic segmentation toolbox and benchmark.
- MMAction2: OpenMMLab's next-generation action understanding toolbox and benchmark.
- MMTracking: OpenMMLab video perception toolbox and benchmark.
- MMPose: OpenMMLab pose estimation toolbox and benchmark.
- MMEditing: OpenMMLab image and video editing toolbox.
- MMOCR: OpenMMLab text detection, recognition and understanding toolbox.
- MMGeneration: OpenMMLab image and video generative models toolbox.